diff --git a/model.py b/model.py index ed35f1f..cd244bb 100644 --- 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)