Skip to content

GitLab

"torchvision/git@developer.sourcefind.cn:OpenDAS/vision.git" did not exist on "d1ab583d0d2df73208e2fc9c4d3a84e969c69b70"
Commit d5347856 authored by Vitor Finotti Ferreira's avatar Vitor Finotti Ferreira Committed by Soumith Chintala
Browse files

Auto calculating return dimension of squeezenet forward method (#884)

parent 0c575ace
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
torchvision/models/squeezenet.py
View file @ d5347856
...@@ -99,7 +99,7 @@ class SqueezeNet(nn.Module): ...@@ -99,7 +99,7 @@ class SqueezeNet(nn.Module):
def forward(self, x): def forward(self, x):
x = self.features(x) x = self.features(x)
x = self.classifier(x) x = self.classifier(x)
return x.view(x.size(0), self.num_classes) return x.view(x.size(0), -1)
def _squeezenet(version, pretrained, progress, **kwargs): def _squeezenet(version, pretrained, progress, **kwargs):
......
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