Skip to content

Commit

Permalink
[fbsync] Add regnet_y_128gf from SWAG (#5732)
Browse files Browse the repository at this point in the history
Summary:
* Add regnet_y_128gh_swag weight

* Add default weight for regnet_y_128gf

* Add the accuracy from experiments

Reviewed By: NicolasHug

Differential Revision: D35393169

fbshipit-source-id: 2086ec2a7310cc83b4a294403d7e380f3f338e50
  • Loading branch information
datumbox authored and facebook-github-bot committed Apr 6, 2022
1 parent db4cc0b commit ab524a8
Showing 1 changed file with 13 additions and 2 deletions.
15 changes: 13 additions & 2 deletions torchvision/models/regnet.py
Original file line number Diff line number Diff line change
Expand Up @@ -629,8 +629,19 @@ class RegNet_Y_32GF_Weights(WeightsEnum):


class RegNet_Y_128GF_Weights(WeightsEnum):
# weights are not available yet.
pass
IMAGENET1K_SWAG_V1 = Weights(
url="https://download.pytorch.org/models/regnet_y_128gf_swag-c8ce3e52.pth",
transforms=partial(
ImageClassification, crop_size=384, resize_size=384, interpolation=InterpolationMode.BICUBIC
),
meta={
**_COMMON_SWAG_META,
"num_params": 644812894,
"acc@1": 88.228,
"acc@5": 98.682,
},
)
DEFAULT = IMAGENET1K_SWAG_V1


class RegNet_X_400MF_Weights(WeightsEnum):
Expand Down

0 comments on commit ab524a8

Please sign in to comment.