Update model.py

leaderj1001 · Oct 19, 2020 · 782e5e8 · 782e5e8
1 parent 7434f74
commit 782e5e8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/model.py b/model.py
@@ -119,7 +119,7 @@ def __init__(self, block, num_blocks, num_classes=1000):
 
         self.conv1 = nn.Conv2d(3, 64, kernel_size=7, stride=2, padding=3, bias=False)
         self.bn1 = nn.BatchNorm2d(64)
-        self.layer1 = self._make_layer(block, 64, num_blocks[0], stride=1)
+        self.layer1 = self._make_layer(block, 64, num_blocks[0], stride=2)
         self.layer2 = self._make_layer(block, 128, num_blocks[1], stride=2)
         self.layer3 = self._make_layer(block, 256, num_blocks[2], stride=2)
         self.layer4 = self._make_layer(block, 512, num_blocks[3], stride=2)