From 6e4bbdb7037ae83ceb80de3ba8b36444bc2e51dc Mon Sep 17 00:00:00 2001 From: Tal Ben-Nun Date: Sun, 16 Jan 2022 17:29:45 +0100 Subject: [PATCH 1/2] Update ResNet-50 accuracy with Repeated Augmentations --- 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 334a3ac7977..caed6e62b3a 100644 --- a/torchvision/prototype/models/resnet.py +++ b/torchvision/prototype/models/resnet.py @@ -115,9 +115,9 @@ class ResNet50_Weights(WeightsEnum): "architecture": "ResNet", "publication_year": 2015, "num_params": 25557032, - "recipe": "https://github.com/pytorch/vision/issues/3995#new-recipe-with-fixres", - "acc@1": 80.674, - "acc@5": 95.166, + "recipe": "https://github.com/pytorch/vision/issues/3995#issuecomment-1013906621", + "acc@1": 80.858, + "acc@5": 95.434, }, ) default = ImageNet1K_V2 From f027050af19e47d092347d3082814b31cf677a35 Mon Sep 17 00:00:00 2001 From: Vasilis Vryniotis Date: Mon, 17 Jan 2022 10:00:47 +0000 Subject: [PATCH 2/2] Update the ResNet50 weights URL. --- torchvision/prototype/models/resnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchvision/prototype/models/resnet.py b/torchvision/prototype/models/resnet.py index caed6e62b3a..96f1cd02a28 100644 --- a/torchvision/prototype/models/resnet.py +++ b/torchvision/prototype/models/resnet.py @@ -108,7 +108,7 @@ class ResNet50_Weights(WeightsEnum): }, ) ImageNet1K_V2 = Weights( - url="https://download.pytorch.org/models/resnet50-f46c3f97.pth", + url="https://download.pytorch.org/models/resnet50-11ad3fa6.pth", transforms=partial(ImageNetEval, crop_size=224, resize_size=232), meta={ **_COMMON_META,