Fixed the input image size for the resnet18

yunjey · web-flow · commit f73f67e3100a · 2018-02-15T22:48:51.000+09:00
diff --git a/tutorials/01-basics/pytorch_basics/main.py b/tutorials/01-basics/pytorch_basics/main.py
@@ -150,7 +150,7 @@ def __len__(self):
 resnet.fc = nn.Linear(resnet.fc.in_features, 100)  # 100 is for example.
 
 # For test.
-images = Variable(torch.randn(10, 3, 256, 256))
+images = Variable(torch.randn(10, 3, 224, 224))
 outputs = resnet(images)
 print (outputs.size())   # (10, 100)
 
@@ -162,4 +162,4 @@ def __len__(self):
 
 # Save and load only the model parameters(recommended).
 torch.save(resnet.state_dict(), 'params.pkl')
-resnet.load_state_dict(torch.load('params.pkl'))
+resnet.load_state_dict(torch.load('params.pkl'))
