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

rm unused

parent 64959bd7
import logging
import math
import torch.nn as nn
......@@ -40,7 +39,7 @@ class AlexNet(nn.Module):
nn.ReLU(inplace=True),
nn.Dropout(),
nn.Linear(4096, 4096),
nn.ReLU(inplace=True), # caffe has dropout
nn.ReLU(inplace=True),
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