diff --git a/torchvision/prototype/models/resnet.py b/torchvision/prototype/models/resnet.py index 2d8cc71e526..5f289c755c0 100644 --- a/torchvision/prototype/models/resnet.py +++ b/torchvision/prototype/models/resnet.py @@ -219,6 +219,16 @@ class WideResNet101_2Weights(Weights): "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: