From 5f7fa2502e34cffd40064e6abb1e40b9250aaf84 Mon Sep 17 00:00:00 2001 From: Kai Zhang Date: Thu, 30 Sep 2021 23:48:47 +0000 Subject: [PATCH 1/3] add best weights and x_1_6, x_3_2, y_1_6, y_3_2, y_32 weights --- docs/source/models.rst | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) diff --git a/docs/source/models.rst b/docs/source/models.rst index ef9c326ade4..962161159ae 100644 --- a/docs/source/models.rst +++ b/docs/source/models.rst @@ -209,12 +209,17 @@ EfficientNet-B4 83.384 96.594 EfficientNet-B5 83.444 96.628 EfficientNet-B6 84.008 96.916 EfficientNet-B7 84.122 96.908 -regnet_x_400mf 72.834 90.950 -regnet_x_800mf 75.190 92.418 -regnet_x_8gf 79.324 94.694 -regnet_y_400mf 74.024 91.680 +regnet_x_400mf 72.834 90.950 +regnet_x_800mf 75.212 92.348 +regnet_x_1_6gf 77.040 93.440 +regnet_x_3_2gf 78.364 93.992 +regnet_x_8gf 79.344 94.686 +regnet_y_400mf 74.046 91.716 regnet_y_800mf 76.420 93.136 -regnet_y_8gf 79.966 95.100 +regnet_y_1_6gf 77.950 93.966 +regnet_y_3_2gf 78.948 94.576 +regnet_y_8gf 80.032 95.048 +regnet_y_32gf 80.878 95.340 ================================ ============= ============= From 6ac8ae049cdc7a385a669222cdeb017d6bc2252d Mon Sep 17 00:00:00 2001 From: Kai Zhang Date: Thu, 30 Sep 2021 23:53:45 +0000 Subject: [PATCH 2/3] add best weights and x_1_6, x_3_2, y_1_6, y_3_2, y_32 weights --- docs/source/models.rst | 5 +++++ torchvision/models/regnet.py | 13 +++++++++---- 2 files changed, 14 insertions(+), 4 deletions(-) diff --git a/docs/source/models.rst b/docs/source/models.rst index 962161159ae..d7454bb0b9a 100644 --- a/docs/source/models.rst +++ b/docs/source/models.rst @@ -111,9 +111,14 @@ These can be constructed by passing ``pretrained=True``: efficientnet_b7 = models.efficientnet_b7(pretrained=True) regnet_y_400mf = models.regnet_y_400mf(pretrained=True) regnet_y_800mf = models.regnet_y_800mf(pretrained=True) + regnet_y_1_6gf = models.regnet_y_1_6gf(pretrained=True) + regnet_y_3_2gf = models.regnet_y_3_2gf(pretrained=True) regnet_y_8gf = models.regnet_y_8gf(pretrained=True) + regnet_y_32gf = models.regnet_y_32gf(pretrained=True) regnet_x_400mf = models.regnet_x_400mf(pretrained=True) regnet_x_800mf = models.regnet_x_800mf(pretrained=True) + regnet_x_1_6gf = models.regnet_x_1_6gf(pretrained=True) + regnet_x_3_2gf = models.regnet_x_3_2gf(pretrained=True) regnet_x_8gf = models.regnet_x_8gf(pretrained=True) Instancing a pre-trained model will download its weights to a cache directory. diff --git a/torchvision/models/regnet.py b/torchvision/models/regnet.py index 0bd89f7799f..0c0311ed107 100644 --- a/torchvision/models/regnet.py +++ b/torchvision/models/regnet.py @@ -23,12 +23,17 @@ model_urls = { - "regnet_y_400mf": "https://download.pytorch.org/models/regnet_y_400mf-540e987b.pth", + "regnet_y_400mf": "https://download.pytorch.org/models/regnet_y_400mf-c65dace8.pth", "regnet_y_800mf": "https://download.pytorch.org/models/regnet_y_800mf-1b27b58c.pth", - "regnet_y_8gf": "https://download.pytorch.org/models/regnet_y_8gf-49ff86b5.pth", + "regnet_y_1_6gf": "https://download.pytorch.org/models/regnet_y_1_6gf-b11a554e.pth", + "regnet_y_3_2gf": "https://download.pytorch.org/models/regnet_y_3_2gf-b5a9779c.pth", + "regnet_y_8gf": "https://download.pytorch.org/models/regnet_y_8gf-d0d0e4a8.pth", + "regnet_y_32gf": "https://download.pytorch.org/models/regnet_y_32gf-4dee3f7a.pth", "regnet_x_400mf": "https://download.pytorch.org/models/regnet_x_400mf-adf1edd5.pth", - "regnet_x_800mf": "https://download.pytorch.org/models/regnet_x_800mf-5cb79b7e.pth", - "regnet_x_8gf": "https://download.pytorch.org/models/regnet_x_8gf-4c4e575e.pth", + "regnet_x_800mf": "https://download.pytorch.org/models/regnet_x_800mf-ad17e45c.pth", + "regnet_x_1_6gf": "https://download.pytorch.org/models/regnet_x_1_6gf-e3633e7f.pth", + "regnet_x_3_2gf": "https://download.pytorch.org/models/regnet_x_3_2gf-f342aeae.pth", + "regnet_x_8gf": "https://download.pytorch.org/models/regnet_x_8gf-03ceed89.pth", } From 17aec7d13ac47e508f017acfb39e61f669f5a521 Mon Sep 17 00:00:00 2001 From: Kai Zhang Date: Mon, 4 Oct 2021 06:57:21 +0000 Subject: [PATCH 3/3] add weights for x_16gf, x_32gf, y_16gf --- docs/source/models.rst | 6 ++++++ torchvision/models/regnet.py | 3 +++ 2 files changed, 9 insertions(+) diff --git a/docs/source/models.rst b/docs/source/models.rst index d7454bb0b9a..4ac582a33a2 100644 --- a/docs/source/models.rst +++ b/docs/source/models.rst @@ -114,12 +114,15 @@ These can be constructed by passing ``pretrained=True``: regnet_y_1_6gf = models.regnet_y_1_6gf(pretrained=True) regnet_y_3_2gf = models.regnet_y_3_2gf(pretrained=True) regnet_y_8gf = models.regnet_y_8gf(pretrained=True) + regnet_y_16gf = models.regnet_y_16gf(pretrained=True) regnet_y_32gf = models.regnet_y_32gf(pretrained=True) regnet_x_400mf = models.regnet_x_400mf(pretrained=True) regnet_x_800mf = models.regnet_x_800mf(pretrained=True) regnet_x_1_6gf = models.regnet_x_1_6gf(pretrained=True) regnet_x_3_2gf = models.regnet_x_3_2gf(pretrained=True) regnet_x_8gf = models.regnet_x_8gf(pretrained=True) + regnet_x_16gf = models.regnet_x_16gf(pretrainedTrue) + regnet_x_32gf = models.regnet_x_32gf(pretrained=True) Instancing a pre-trained model will download its weights to a cache directory. This directory can be set using the `TORCH_MODEL_ZOO` environment variable. See @@ -219,11 +222,14 @@ regnet_x_800mf 75.212 92.348 regnet_x_1_6gf 77.040 93.440 regnet_x_3_2gf 78.364 93.992 regnet_x_8gf 79.344 94.686 +regnet_x_16gf 80.058 94.944 +regnet_x_32gf 80.622 95.248 regnet_y_400mf 74.046 91.716 regnet_y_800mf 76.420 93.136 regnet_y_1_6gf 77.950 93.966 regnet_y_3_2gf 78.948 94.576 regnet_y_8gf 80.032 95.048 +regnet_y_16gf 80.424 95.240 regnet_y_32gf 80.878 95.340 ================================ ============= ============= diff --git a/torchvision/models/regnet.py b/torchvision/models/regnet.py index 0c0311ed107..c7694625d52 100644 --- a/torchvision/models/regnet.py +++ b/torchvision/models/regnet.py @@ -28,12 +28,15 @@ "regnet_y_1_6gf": "https://download.pytorch.org/models/regnet_y_1_6gf-b11a554e.pth", "regnet_y_3_2gf": "https://download.pytorch.org/models/regnet_y_3_2gf-b5a9779c.pth", "regnet_y_8gf": "https://download.pytorch.org/models/regnet_y_8gf-d0d0e4a8.pth", + "regnet_y_16gf": "https://download.pytorch.org/models/regnet_y_16gf-9e6ed7dd.pth", "regnet_y_32gf": "https://download.pytorch.org/models/regnet_y_32gf-4dee3f7a.pth", "regnet_x_400mf": "https://download.pytorch.org/models/regnet_x_400mf-adf1edd5.pth", "regnet_x_800mf": "https://download.pytorch.org/models/regnet_x_800mf-ad17e45c.pth", "regnet_x_1_6gf": "https://download.pytorch.org/models/regnet_x_1_6gf-e3633e7f.pth", "regnet_x_3_2gf": "https://download.pytorch.org/models/regnet_x_3_2gf-f342aeae.pth", "regnet_x_8gf": "https://download.pytorch.org/models/regnet_x_8gf-03ceed89.pth", + "regnet_x_16gf": "https://download.pytorch.org/models/regnet_x_16gf-2007eb11.pth", + "regnet_x_32gf": "https://download.pytorch.org/models/regnet_x_32gf-9d47f8d0.pth", }