Update squeezenet.py (#172)

skrish13 · soumith · commit f4c4d6cb3fd9 · 2017-05-21T12:52:52.000-04:00
changed a small mistake.
diff --git a/torchvision/models/squeezenet.py b/torchvision/models/squeezenet.py
@@ -78,7 +78,7 @@ def __init__(self, version=1.0, num_classes=1000):
                 Fire(512, 64, 256, 256),
             )
         # 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(
             nn.Dropout(p=0.5),
             final_conv,

Original file line number	Diff line number	Diff line change
`@@ -78,7 +78,7 @@ def __init__(self, version=1.0, num_classes=1000):`
`78`	`78`	`Fire(512, 64, 256, 256),`
`79`	`79`	`)`
`80`	`80`	`# Final convolution is initialized differently form the rest`
`81`		`- final_conv = nn.Conv2d(512, num_classes, kernel_size=1)`
	`81`	`+ final_conv = nn.Conv2d(512, self.num_classes, kernel_size=1)`
`82`	`82`	`self.classifier = nn.Sequential(`
`83`	`83`	`nn.Dropout(p=0.5),`
`84`	`84`	`final_conv,`