From c7003de3e800d93e5eaa852b487fde18418d5b23 Mon Sep 17 00:00:00 2001 From: Yosua Michael Maranatha Date: Mon, 4 Apr 2022 11:04:04 +0100 Subject: [PATCH 1/3] Add regnet_y_128gh_swag weight --- torchvision/models/regnet.py | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/torchvision/models/regnet.py b/torchvision/models/regnet.py index 0748ee0460d..05ce6282a63 100644 --- a/torchvision/models/regnet.py +++ b/torchvision/models/regnet.py @@ -629,7 +629,19 @@ class RegNet_Y_32GF_Weights(WeightsEnum): class RegNet_Y_128GF_Weights(WeightsEnum): - # weights are not available yet. + 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, + # Still mock + "acc@1": 88.23, + "acc@5": 98.69, + }, + ) pass From 6da531d5d3eeed9a26b76e2d7499c5aba4291d9b Mon Sep 17 00:00:00 2001 From: Yosua Michael Maranatha Date: Mon, 4 Apr 2022 11:04:53 +0100 Subject: [PATCH 2/3] Add default weight for regnet_y_128gf --- torchvision/models/regnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchvision/models/regnet.py b/torchvision/models/regnet.py index 05ce6282a63..34b2970073e 100644 --- a/torchvision/models/regnet.py +++ b/torchvision/models/regnet.py @@ -642,7 +642,7 @@ class RegNet_Y_128GF_Weights(WeightsEnum): "acc@5": 98.69, }, ) - pass + DEFAULT = IMAGENET1K_SWAG_V1 class RegNet_X_400MF_Weights(WeightsEnum): From 278f0f7d9f916531b9120401decf5bec938c6046 Mon Sep 17 00:00:00 2001 From: Yosua Michael Maranatha Date: Mon, 4 Apr 2022 13:27:37 +0100 Subject: [PATCH 3/3] Add the accuracy from experiments --- torchvision/models/regnet.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/torchvision/models/regnet.py b/torchvision/models/regnet.py index 34b2970073e..ce1885e4a04 100644 --- a/torchvision/models/regnet.py +++ b/torchvision/models/regnet.py @@ -637,9 +637,8 @@ class RegNet_Y_128GF_Weights(WeightsEnum): meta={ **_COMMON_SWAG_META, "num_params": 644812894, - # Still mock - "acc@1": 88.23, - "acc@5": 98.69, + "acc@1": 88.228, + "acc@5": 98.682, }, ) DEFAULT = IMAGENET1K_SWAG_V1