From 87eb94509c2b9c0daeaa5aa59d134beeca990b77 Mon Sep 17 00:00:00 2001 From: Vasilis Vryniotis Date: Mon, 1 Nov 2021 12:16:40 +0000 Subject: [PATCH] Add new weights for ResNet101 and ResNet151. --- torchvision/prototype/models/resnet.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/torchvision/prototype/models/resnet.py b/torchvision/prototype/models/resnet.py index b6259b4a803..d777a493649 100644 --- a/torchvision/prototype/models/resnet.py +++ b/torchvision/prototype/models/resnet.py @@ -114,6 +114,16 @@ class ResNet101Weights(Weights): "acc@5": 93.546, }, ) + ImageNet1K_RefV2 = WeightEntry( + url="https://download.pytorch.org/models/resnet101-b641f3a9.pth", + transforms=partial(ImageNetEval, crop_size=224, resize_size=232), + meta={ + **_common_meta, + "recipe": "https://github.com/pytorch/vision/issues/3995", + "acc@1": 81.728, + "acc@5": 95.670, + }, + ) class ResNet152Weights(Weights): @@ -127,6 +137,16 @@ class ResNet152Weights(Weights): "acc@5": 94.046, }, ) + ImageNet1K_RefV2 = WeightEntry( + url="https://download.pytorch.org/models/resnet152-089c0848.pth", + transforms=partial(ImageNetEval, crop_size=224, resize_size=232), + meta={ + **_common_meta, + "recipe": "https://github.com/pytorch/vision/issues/3995", + "acc@1": 82.042, + "acc@5": 95.926, + }, + ) class ResNeXt50_32x4dWeights(Weights):