Change usage of "view" to "reshape" in resnet to enable running with mkldnn (#890)

ce18507b · bddppq · Francisco Massa · dc7db17e · ce18507b
Commit ce18507b authored May 07, 2019 by bddppq Committed by Francisco Massa May 07, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

torchvision/models/resnet.py torchvision/models/resnet.py +1 -1

No files found.
--- a/torchvision/models/resnet.py
+++ b/torchvision/models/resnet.py
@@ -198,7 +198,7 @@ class ResNet(nn.Module):
        x = self.layer4(x)
        x = self.avgpool(x)
-        x = x.view(x.size(0), -1)
+        x = x.reshape(x.size(0), -1)
        x = self.fc(x)
        return x