From 7a37cc5d0a8ca19dc245f5cd0ac84c0629f5035f Mon Sep 17 00:00:00 2001 From: Kai Zhang <1006921+kazhang@users.noreply.github.com> Date: Mon, 4 Oct 2021 21:46:03 -0700 Subject: [PATCH] Update Regnet model weights (#4530) * add best weights and x_1_6, x_3_2, y_1_6, y_3_2, y_32 weights * add best weights and x_1_6, x_3_2, y_1_6, y_3_2, y_32 weights * add weights for x_16gf, x_32gf, y_16gf --- docs/source/models.rst | 26 +++++++++++++++++++++----- torchvision/models/regnet.py | 16 ++++++++++++---- 2 files changed, 33 insertions(+), 9 deletions(-) diff --git a/docs/source/models.rst b/docs/source/models.rst index ef9c326ade4..4ac582a33a2 100644 --- a/docs/source/models.rst +++ b/docs/source/models.rst @@ -111,10 +111,18 @@ 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_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 @@ -209,12 +217,20 @@ 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_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_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_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 0bd89f7799f..c7694625d52 100644 --- a/torchvision/models/regnet.py +++ b/torchvision/models/regnet.py @@ -23,12 +23,20 @@ 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_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-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", + "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", }