diff --git a/efficientnet_pytorch/__init__.py b/efficientnet_pytorch/__init__.py index 4c93560..b8a7aef 100644 --- 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 index f9b59ab..325cffc 100755 --- 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 index 6e35428..03a9077 100644 --- 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 = [