Unverified Commit ed2eefc7 authored by Prabhat Roy's avatar Prabhat Roy Committed by GitHub
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: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
parent 4715e2e5
......@@ -196,6 +196,16 @@ class WideResNet50_2Weights(Weights):
"acc@5": 94.086,
},
)
ImageNet1K_RefV2 = WeightEntry(
url="https://download.pytorch.org/models/wide_resnet50_2-9ba9bcbe.pth",
transforms=partial(ImageNetEval, crop_size=224, resize_size=232),
meta={
**_common_meta,
"recipe": "https://github.com/pytorch/vision/issues/3995",
"acc@1": 81.602,
"acc@5": 95.758,
},
)
class WideResNet101_2Weights(Weights):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment