Unverified Commit 348f75ce authored by Prabhat Roy's avatar Prabhat Roy Committed by GitHub
Browse files

Add new weights for wide_resnet101_2 model (#4906)



* Add new weights for wide_resnet101_2 model

* Updated accuracies
Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
parent a2d240ee
...@@ -219,6 +219,16 @@ class WideResNet101_2Weights(Weights): ...@@ -219,6 +219,16 @@ class WideResNet101_2Weights(Weights):
"acc@5": 94.284, "acc@5": 94.284,
}, },
) )
ImageNet1K_RefV2 = WeightEntry(
url="https://download.pytorch.org/models/wide_resnet101_2-b8680a8c.pth",
transforms=partial(ImageNetEval, crop_size=224, resize_size=232),
meta={
**_common_meta,
"recipe": "https://github.com/pytorch/vision/issues/3995",
"acc@1": 82.492,
"acc@5": 96.110,
},
)
def resnet18(weights: Optional[ResNet18Weights] = None, progress: bool = True, **kwargs: Any) -> ResNet: def resnet18(weights: Optional[ResNet18Weights] = None, progress: bool = True, **kwargs: Any) -> ResNet:
......
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