Unverified Commit f923aeb2 authored by Tal Ben-Nun's avatar Tal Ben-Nun Committed by GitHub
Browse files

Update ResNet-50 accuracy with Repeated Augmentation (#5201)



* Update ResNet-50 accuracy with Repeated Augmentations

* Update the ResNet50 weights URL.
Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
parent 09c5ddd6
...@@ -108,16 +108,16 @@ class ResNet50_Weights(WeightsEnum): ...@@ -108,16 +108,16 @@ class ResNet50_Weights(WeightsEnum):
}, },
) )
ImageNet1K_V2 = Weights( 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), transforms=partial(ImageNetEval, crop_size=224, resize_size=232),
meta={ meta={
**_COMMON_META, **_COMMON_META,
"architecture": "ResNet", "architecture": "ResNet",
"publication_year": 2015, "publication_year": 2015,
"num_params": 25557032, "num_params": 25557032,
"recipe": "https://github.com/pytorch/vision/issues/3995#new-recipe-with-fixres", "recipe": "https://github.com/pytorch/vision/issues/3995#issuecomment-1013906621",
"acc@1": 80.674, "acc@1": 80.858,
"acc@5": 95.166, "acc@5": 95.434,
}, },
) )
default = ImageNet1K_V2 default = ImageNet1K_V2
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment