From 6529f6a32c43f828af7a544064c6d9ea3e49e76c Mon Sep 17 00:00:00 2001 From: Prabhat Roy Date: Wed, 10 Nov 2021 22:02:27 +0000 Subject: [PATCH 1/2] Add new weights for wide_resnet101_2 model --- torchvision/prototype/models/resnet.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/torchvision/prototype/models/resnet.py b/torchvision/prototype/models/resnet.py index ff82d5e3172..519a4a6bc07 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.498, + "acc@5": 96.114, + }, + ) def resnet18(weights: Optional[ResNet18Weights] = None, progress: bool = True, **kwargs: Any) -> ResNet: From e074b3837d4461e305bc38ebdcad1b891be4361c Mon Sep 17 00:00:00 2001 From: Prabhat Roy Date: Wed, 10 Nov 2021 23:33:24 +0000 Subject: [PATCH 2/2] Updated accuracies --- torchvision/prototype/models/resnet.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/torchvision/prototype/models/resnet.py b/torchvision/prototype/models/resnet.py index 519a4a6bc07..b8a2717ca1e 100644 --- a/torchvision/prototype/models/resnet.py +++ b/torchvision/prototype/models/resnet.py @@ -225,8 +225,8 @@ class WideResNet101_2Weights(Weights): meta={ **_common_meta, "recipe": "https://github.com/pytorch/vision/issues/3995", - "acc@1": 82.498, - "acc@5": 96.114, + "acc@1": 82.492, + "acc@5": 96.110, }, )