Skip to content

Commit ed2eefc

Browse files
Add new weights for wide_resnet50_2 model (#4862)
* Updated wide_resnet50_2 model weights * Update Accs with batch-size 1. * Update checkpoint and accs. Co-authored-by: Vasilis Vryniotis <[email protected]>
1 parent 4715e2e commit ed2eefc

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)