@@ -54,7 +54,7 @@ class Alexnet(nn.Module):
def forward(self, x):
x = self.features(x)
- x = x.view(x.size(0), -1)
+ x = x.reshape(x.size(0), -1)
x = self.classifier(x)
return x
