Skip to content

Commit

Permalink
[fbsync] Update ResNet-50 accuracy with Repeated Augmentation (#5201)
Browse files Browse the repository at this point in the history
Summary:
* Update ResNet-50 accuracy with Repeated Augmentations

* Update the ResNet50 weights URL.

Reviewed By: datumbox, NicolasHug

Differential Revision: D33655256

fbshipit-source-id: 230c8914e4784f84a57f1b89f4f279c34ba8fc3b

Co-authored-by: Vasilis Vryniotis <[email protected]>
  • Loading branch information
2 people authored and facebook-github-bot committed Jan 19, 2022
1 parent bd89978 commit c631b40
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions torchvision/prototype/models/resnet.py
Original file line number Diff line number Diff line change
Expand Up @@ -108,16 +108,16 @@ 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,
"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
Expand Down

0 comments on commit c631b40

Please sign in to comment.