Commit 8332ddbc authored by yl-1993's avatar yl-1993
Browse files

rm unused

parent 64959bd7
import logging import logging
import math
import torch.nn as nn import torch.nn as nn
...@@ -40,7 +39,7 @@ class AlexNet(nn.Module): ...@@ -40,7 +39,7 @@ class AlexNet(nn.Module):
nn.ReLU(inplace=True), nn.ReLU(inplace=True),
nn.Dropout(), nn.Dropout(),
nn.Linear(4096, 4096), nn.Linear(4096, 4096),
nn.ReLU(inplace=True), # caffe has dropout nn.ReLU(inplace=True),
nn.Linear(4096, num_classes), nn.Linear(4096, num_classes),
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment