From f73f67e3100ad9f7bf435a156a163a10a0f48cf2 Mon Sep 17 00:00:00 2001 From: Yunjey Choi Date: Thu, 15 Feb 2018 22:48:51 +0900 Subject: [PATCH] Fixed the input image size for the resnet18 --- tutorials/01-basics/pytorch_basics/main.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tutorials/01-basics/pytorch_basics/main.py b/tutorials/01-basics/pytorch_basics/main.py index 17a5070..153d23a 100644 --- a/tutorials/01-basics/pytorch_basics/main.py +++ b/tutorials/01-basics/pytorch_basics/main.py @@ -150,7 +150,7 @@ for param in resnet.parameters(): 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 @@ model = torch.load('model.pkl') # Save and load only the model parameters(recommended). torch.save(resnet.state_dict(), 'params.pkl') -resnet.load_state_dict(torch.load('params.pkl')) \ No newline at end of file +resnet.load_state_dict(torch.load('params.pkl'))