Merge pull request yunjey#141 from keineahnung2345/fix-resnet

[Bug fix] redundant layers in ResNet

Author: yunjey

tutorials/02-intermediate/deep_residual_network/main.py

@@ -82,8 +82,8 @@ def __init__(self, block, layers, num_classes=10):
         self.bn = nn.BatchNorm2d(16)
         self.relu = nn.ReLU(inplace=True)
         self.layer1 = self.make_layer(block, 16, layers[0])
-        self.layer2 = self.make_layer(block, 32, layers[0], 2)
-        self.layer3 = self.make_layer(block, 64, layers[1], 2)
+        self.layer2 = self.make_layer(block, 32, layers[1], 2)
+        self.layer3 = self.make_layer(block, 64, layers[2], 2)
         self.avg_pool = nn.AvgPool2d(8)
         self.fc = nn.Linear(64, num_classes)
 
@@ -112,7 +112,7 @@ def forward(self, x):
         out = self.fc(out)
         return out
 
-model = ResNet(ResidualBlock, [2, 2, 2, 2]).to(device)
+model = ResNet(ResidualBlock, [2, 2, 2]).to(device)
 
 
 # Loss and optimizer
@@ -166,4 +166,4 @@ def update_lr(optimizer, lr):
     print('Accuracy of the model on the test images: {} %'.format(100 * correct / total))
 
 # Save the model checkpoint
-torch.save(model.state_dict(), 'resnet.ckpt')
+torch.save(model.state_dict(), 'resnet.ckpt')