From 2a672ceef2c8aac56572c7cc1e8500a044dd7444 Mon Sep 17 00:00:00 2001
From: Vasilis Vryniotis <vvryniotis@fb.com>
Date: Tue, 2 Nov 2021 17:14:21 +0000
Subject: [PATCH] Add new weights for ResNeXt50_32x4d.

---
 torchvision/prototype/models/resnet.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/torchvision/prototype/models/resnet.py b/torchvision/prototype/models/resnet.py
index c4330131201..764ea49641e 100644
--- 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):