From 483a798bf0ff0c39838d828509b4857a232f7dfe Mon Sep 17 00:00:00 2001 From: Prabhat Roy Date: Thu, 4 Nov 2021 19:07:53 +0000 Subject: [PATCH 1/3] Updated wide_resnet50_2 model weights --- 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 764ea49641e..b2d8a47d3c8 100644 --- a/torchvision/prototype/models/resnet.py +++ b/torchvision/prototype/models/resnet.py @@ -196,6 +196,16 @@ class WideResNet50_2Weights(Weights): "acc@5": 94.086, }, ) + ImageNet1K_RefV2 = WeightEntry( + url="https://download.pytorch.org/models/wide_resnet50_2-8be09fd8.pth", + transforms=partial(ImageNetEval, crop_size=224, resize_size=232), + meta={ + **_common_meta, + "recipe": "https://github.com/pytorch/vision/issues/3995", + "acc@1": 81.604, + "acc@5": 95.756, + }, + ) class WideResNet101_2Weights(Weights): From 5690476ba42b5b1543b16dd95d488968a9e086de Mon Sep 17 00:00:00 2001 From: Vasilis Vryniotis Date: Fri, 5 Nov 2021 09:59:32 +0000 Subject: [PATCH 2/3] Update Accs with batch-size 1. --- 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 b2d8a47d3c8..f3cd624afde 100644 --- a/torchvision/prototype/models/resnet.py +++ b/torchvision/prototype/models/resnet.py @@ -202,8 +202,8 @@ class WideResNet50_2Weights(Weights): meta={ **_common_meta, "recipe": "https://github.com/pytorch/vision/issues/3995", - "acc@1": 81.604, - "acc@5": 95.756, + "acc@1": 81.514, + "acc@5": 95.686, }, ) From 157b9e7ae615747b6f81394710304d1d8bc89a79 Mon Sep 17 00:00:00 2001 From: Vasilis Vryniotis Date: Fri, 5 Nov 2021 13:06:55 +0000 Subject: [PATCH 3/3] Update checkpoint and accs. --- torchvision/prototype/models/resnet.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/torchvision/prototype/models/resnet.py b/torchvision/prototype/models/resnet.py index f3cd624afde..ff82d5e3172 100644 --- a/torchvision/prototype/models/resnet.py +++ b/torchvision/prototype/models/resnet.py @@ -197,13 +197,13 @@ class WideResNet50_2Weights(Weights): }, ) ImageNet1K_RefV2 = WeightEntry( - url="https://download.pytorch.org/models/wide_resnet50_2-8be09fd8.pth", + 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.514, - "acc@5": 95.686, + "acc@1": 81.602, + "acc@5": 95.758, }, )