Add new weights for ResNeXt50_32x4d. (#4836)

c5fb79f8 · Vasilis Vryniotis · GitHub · 9b7d5d52 · c5fb79f8
Unverified Commit c5fb79f8 authored Nov 02, 2021 by Vasilis Vryniotis Committed by GitHub Nov 02, 2021
Show whitespace changes
Inline Side-by-side

Showing with 10 additions and 0 deletions

torchvision/prototype/models/resnet.py torchvision/prototype/models/resnet.py +10 -0

No files found.
--- a/torchvision/prototype/models/resnet.py
+++ b/torchvision/prototype/models/resnet.py
@@ -160,6 +160,16 @@ class ResNeXt50_32x4dWeights(Weights):
            "acc@5": 93.698,
        },
    )
+    ImageNet1K_RefV2 = WeightEntry(
+        url="https://download.pytorch.org/models/resnext50_32x4d-b260af35.pth",
+        transforms=partial(ImageNetEval, crop_size=224, resize_size=232),
+        meta={
+            **_common_meta,
+            "recipe": "https://github.com/pytorch/vision/issues/3995",
+            "acc@1": 81.116,
+            "acc@5": 95.478,
+        },
+    )
 class ResNeXt101_32x8dWeights(Weights):