Skip to content

GitLab

Commit f4c4d6cb authored by Sri Krishna's avatar Sri Krishna Committed by Soumith Chintala
Browse files

Update squeezenet.py (#172) 

changed a small mistake.
parent cbb05c59
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
torchvision/models/squeezenet.py
View file @ f4c4d6cb
...@@ -78,7 +78,7 @@ class SqueezeNet(nn.Module): ...@@ -78,7 +78,7 @@ class SqueezeNet(nn.Module):
Fire(512, 64, 256, 256), Fire(512, 64, 256, 256),
) )
# Final convolution is initialized differently form the rest # Final convolution is initialized differently form the rest
final_conv = nn.Conv2d(512, num_classes, kernel_size=1) final_conv = nn.Conv2d(512, self.num_classes, kernel_size=1)
self.classifier = nn.Sequential( self.classifier = nn.Sequential(
nn.Dropout(p=0.5), nn.Dropout(p=0.5),
final_conv, final_conv,
......
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