Skip to content

Commit 5357fc9

Browse files
prabhat00155facebook-github-bot
authored andcommitted
[fbsync] Add new weights for ResNeXt50_32x4d. (#4836)
Reviewed By: kazhang Differential Revision: D32216661 fbshipit-source-id: b3c0dd2c17826d4eae134b1c95f8b2a45c118bd3
1 parent d237d95 commit 5357fc9

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)