algorithm/pytorch-tutorial
1
0
Fork 0
mirror of https://github.com/yunjey/pytorch-tutorial.git synced 2025-07-05 16:36:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fixed the input image size for the resnet18

Browse Source
This commit is contained in:
Yunjey Choi
2018-02-15 22:48:51 +09:00
committed by GitHub
parent 655c6df3ba
commit f73f67e310
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tutorials/01-basics/pytorch_basics/main.py
View File

@ -150,7 +150,7 @@ for param in resnet.parameters():
resnet.fc = nn.Linear(resnet.fc.in_features, 100) # 100 is for example. resnet.fc = nn.Linear(resnet.fc.in_features, 100) # 100 is for example.
# For test. # For test.
images = Variable(torch.randn(10, 3, 256, 256)) images = Variable(torch.randn(10, 3, 224, 224))
outputs = resnet(images) outputs = resnet(images)
print (outputs.size()) # (10, 100) print (outputs.size()) # (10, 100)