Updated pretrained weight urls

lukemelas · Mar 1, 2020 · e047e4e · e047e4e
1 parent f8d1eb4
commit e047e4e
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 19 deletions.
diff --git a/efficientnet_pytorch/__init__.py b/efficientnet_pytorch/__init__.py
@@ -1,4 +1,4 @@
-__version__ = "0.6.1"
+__version__ = "0.6.2"
 from .model import EfficientNet
 from .utils import (
     GlobalParams,

diff --git a/efficientnet_pytorch/utils.py b/efficientnet_pytorch/utils.py
@@ -296,27 +296,27 @@ def get_model_params(model_name, override_params):
 
 
 url_map = {
-    'efficientnet-b0': 'https://publicmodels.blob.core.windows.net/container/aa/efficientnet-b0-355c32eb.pth',
-    'efficientnet-b1': 'https://publicmodels.blob.core.windows.net/container/aa/efficientnet-b1-f1951068.pth',
-    'efficientnet-b2': 'https://publicmodels.blob.core.windows.net/container/aa/efficientnet-b2-8bb594d6.pth',
-    'efficientnet-b3': 'https://publicmodels.blob.core.windows.net/container/aa/efficientnet-b3-5fb5a3c3.pth',
-    'efficientnet-b4': 'https://publicmodels.blob.core.windows.net/container/aa/efficientnet-b4-6ed6700e.pth',
-    'efficientnet-b5': 'https://publicmodels.blob.core.windows.net/container/aa/efficientnet-b5-b6417697.pth',
-    'efficientnet-b6': 'https://publicmodels.blob.core.windows.net/container/aa/efficientnet-b6-c76e70fd.pth',
-    'efficientnet-b7': 'https://publicmodels.blob.core.windows.net/container/aa/efficientnet-b7-dcc49843.pth',
+    'efficientnet-b0': 'https://storage.googleapis.com/cloud-tpu-checkpoints/efficientnet/pth/effnet/efficientnet-b0-355c32eb.pth',
+    'efficientnet-b1': 'https://storage.googleapis.com/cloud-tpu-checkpoints/efficientnet/pth/effnet/efficientnet-b1-f1951068.pth',
+    'efficientnet-b2': 'https://storage.googleapis.com/cloud-tpu-checkpoints/efficientnet/pth/effnet/efficientnet-b2-8bb594d6.pth',
+    'efficientnet-b3': 'https://storage.googleapis.com/cloud-tpu-checkpoints/efficientnet/pth/effnet/efficientnet-b3-5fb5a3c3.pth',
+    'efficientnet-b4': 'https://storage.googleapis.com/cloud-tpu-checkpoints/efficientnet/pth/effnet/efficientnet-b4-6ed6700e.pth',
+    'efficientnet-b5': 'https://storage.googleapis.com/cloud-tpu-checkpoints/efficientnet/pth/effnet/efficientnet-b5-b6417697.pth',
+    'efficientnet-b6': 'https://storage.googleapis.com/cloud-tpu-checkpoints/efficientnet/pth/effnet/efficientnet-b6-c76e70fd.pth',
+    'efficientnet-b7': 'https://storage.googleapis.com/cloud-tpu-checkpoints/efficientnet/pth/effnet/efficientnet-b7-dcc49843.pth',
 }
 
 
 url_map_advprop = {
-    'efficientnet-b0': 'https://publicmodels.blob.core.windows.net/container/advprop/efficientnet-b0-b64d5a18.pth', 
-    'efficientnet-b1': 'https://publicmodels.blob.core.windows.net/container/advprop/efficientnet-b1-0f3ce85a.pth',
-    'efficientnet-b2': 'https://publicmodels.blob.core.windows.net/container/advprop/efficientnet-b2-6e9d97e5.pth',
-    'efficientnet-b3': 'https://publicmodels.blob.core.windows.net/container/advprop/efficientnet-b3-cdd7c0f4.pth',
-    'efficientnet-b4': 'https://publicmodels.blob.core.windows.net/container/advprop/efficientnet-b4-44fb3a87.pth',
-    'efficientnet-b5': 'https://publicmodels.blob.core.windows.net/container/advprop/efficientnet-b5-86493f6b.pth',
-    'efficientnet-b6': 'https://publicmodels.blob.core.windows.net/container/advprop/efficientnet-b6-ac80338e.pth',
-    'efficientnet-b7': 'https://publicmodels.blob.core.windows.net/container/advprop/efficientnet-b7-4652b6dd.pth',
-    'efficientnet-b8': 'https://publicmodels.blob.core.windows.net/container/advprop/efficientnet-b8-22a8fe65.pth',
+    'efficientnet-b0': 'https://storage.googleapis.com/cloud-tpu-checkpoints/efficientnet/pth/effnetadv/efficientnet-b0-b64d5a18.pth',
+    'efficientnet-b1': 'https://storage.googleapis.com/cloud-tpu-checkpoints/efficientnet/pth/effnetadv/efficientnet-b1-0f3ce85a.pth',
+    'efficientnet-b2': 'https://storage.googleapis.com/cloud-tpu-checkpoints/efficientnet/pth/effnetadv/efficientnet-b2-6e9d97e5.pth',
+    'efficientnet-b3': 'https://storage.googleapis.com/cloud-tpu-checkpoints/efficientnet/pth/effnetadv/efficientnet-b3-cdd7c0f4.pth',
+    'efficientnet-b4': 'https://storage.googleapis.com/cloud-tpu-checkpoints/efficientnet/pth/effnetadv/efficientnet-b4-44fb3a87.pth',
+    'efficientnet-b5': 'https://storage.googleapis.com/cloud-tpu-checkpoints/efficientnet/pth/effnetadv/efficientnet-b5-86493f6b.pth',
+    'efficientnet-b6': 'https://storage.googleapis.com/cloud-tpu-checkpoints/efficientnet/pth/effnetadv/efficientnet-b6-ac80338e.pth',
+    'efficientnet-b7': 'https://storage.googleapis.com/cloud-tpu-checkpoints/efficientnet/pth/effnetadv/efficientnet-b7-4652b6dd.pth',
+    'efficientnet-b8': 'https://storage.googleapis.com/cloud-tpu-checkpoints/efficientnet/pth/effnetadv/efficientnet-b8-22a8fe65.pth',
 }
 
 

diff --git a/setup.py b/setup.py
@@ -18,7 +18,7 @@
 EMAIL = 'lmelaskyriazi@college.harvard.edu'
 AUTHOR = 'Luke'
 REQUIRES_PYTHON = '>=3.5.0'
-VERSION = '0.6.1'
+VERSION = '0.6.2'
 
 # What packages are required for this module to be executed?
 REQUIRED = [