Update model.py

model.py

@@ -228,4 +228,8 @@ class ResNet(nn.Module):
         x = self.fc(x)
         x = torch.flatten(x, 1)
 
-        return x
+        return x
+
+
+def resnet50(num_classes=1000, drop_path_rate=0.2, use_blurpool=False):
+    return ResNet(BottleneckBlock, [3, 4, 6, 3], num_classes, drop_path_rate, use_blurpool)