Skip to content

Commit dbbf6f3

Browse files
prabhat00155datumbox
authored andcommitted
[fbsync] Add new weights for wide_resnet50_2 model (#4862)
Summary: * Updated wide_resnet50_2 model weights * Update Accs with batch-size 1. * Update checkpoint and accs. Reviewed By: kazhang Differential Revision: D32216677 fbshipit-source-id: 3ea09f12d85f88c131bd29ac8482ae420c17d10f Co-authored-by: Vasilis Vryniotis <[email protected]>
1 parent 75b4e15 commit dbbf6f3

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
@@ -196,6 +196,16 @@ class WideResNet50_2Weights(Weights):
196196
"acc@5": 94.086,
197197
},
198198
)
199+
ImageNet1K_RefV2 = WeightEntry(
200+
url="https://download.pytorch.org/models/wide_resnet50_2-9ba9bcbe.pth",
201+
transforms=partial(ImageNetEval, crop_size=224, resize_size=232),
202+
meta={
203+
**_common_meta,
204+
"recipe": "https://github.com/pytorch/vision/issues/3995",
205+
"acc@1": 81.602,
206+
"acc@5": 95.758,
207+
},
208+
)
199209

200210

201211
class WideResNet101_2Weights(Weights):

0 commit comments

Comments
 (0)