Update model.py

leaderj1001 · Feb 23, 2021 · 4850d9d · 4850d9d
1 parent 1ecf9f1
commit 4850d9d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/model.py b/model.py
@@ -140,7 +140,7 @@ def forward(self, x):
         return out
 
 
-def ResNet50(num_classes=1000, resolution=(256, 256)):
+def ResNet50(num_classes=1000, resolution=(224, 224)):
     return ResNet(Bottleneck, [3, 4, 6, 3], num_classes=num_classes, resolution=resolution)