Skip to content

Commit c5fb79f

Browse files
authored
Add new weights for ResNeXt50_32x4d. (#4836)
1 parent 9b7d5d5 commit c5fb79f

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

torchvision/prototype/models/resnet.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -160,6 +160,16 @@ class ResNeXt50_32x4dWeights(Weights):
160160
"acc@5": 93.698,
161161
},
162162
)
163+
ImageNet1K_RefV2 = WeightEntry(
164+
url="https://download.pytorch.org/models/resnext50_32x4d-b260af35.pth",
165+
transforms=partial(ImageNetEval, crop_size=224, resize_size=232),
166+
meta={
167+
**_common_meta,
168+
"recipe": "https://github.com/pytorch/vision/issues/3995",
169+
"acc@1": 81.116,
170+
"acc@5": 95.478,
171+
},
172+
)
163173

164174

165175
class ResNeXt101_32x8dWeights(Weights):

0 commit comments

Comments
 (0)