Skip to content

GitLab

Commit 0d975d6d authored by Allan Wang's avatar Allan Wang Committed by Francisco Massa
Browse files

Fix typo in squeezenet (#841)

parent 2686e1a3
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
torchvision/models/squeezenet.py
View file @ 0d975d6d
...@@ -76,7 +76,7 @@ class SqueezeNet(nn.Module): ...@@ -76,7 +76,7 @@ class SqueezeNet(nn.Module):
Fire(384, 64, 256, 256), Fire(384, 64, 256, 256),
Fire(512, 64, 256, 256), Fire(512, 64, 256, 256),
) )
# Final convolution is initialized differently form the rest # Final convolution is initialized differently from the rest
final_conv = nn.Conv2d(512, self.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),
......
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