Update ResNeSt Model Weights Step 1 (#156)

* Update resnest.py * Update model_store.py * Update ablation.py
zhanghang1989 · May 18, 2021 · 1dfb3e8 · 1dfb3e8
1 parent a6bde39
commit 1dfb3e8
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/resnest/gluon/model_store.py b/resnest/gluon/model_store.py
@@ -29,8 +29,8 @@
     ('7318153ddb5e542a20cc6c58192f3c6209cff9ed', 'resnest50_fast_1s4x24d'),
     ]}
 
-encoding_repo_url = 'https://s3.us-west-1.wasabisys.com/resnest/'
-_url_format = '{repo_url}gluon/{file_name}.zip'
+encoding_repo_url = 'https://github.com/zhanghang1989/ResNeSt/releases/download/weights_step2'
+_url_format = '{repo_url}/{file_name}.zip'
 
 def short_hash(name):
     if name not in _model_sha1:

diff --git a/resnest/torch/models/ablation.py b/resnest/torch/models/ablation.py
@@ -14,7 +14,7 @@
            'resnest50_fast_1s2x40d', 'resnest50_fast_2s2x40d', 'resnest50_fast_4s2x40d',
            'resnest50_fast_1s4x24d']
 
-_url_format = 'https://s3.us-west-1.wasabisys.com/resnest/torch/{}-{}.pth'
+_url_format = 'https://github.com/zhanghang1989/ResNeSt/releases/download/weights_step1/{}-{}.pth'
 
 _model_sha256 = {name: checksum for checksum, name in [
     ('d8fbf808', 'resnest50_fast_1s1x64d'),

diff --git a/resnest/torch/models/resnest.py b/resnest/torch/models/resnest.py
@@ -13,7 +13,7 @@
 __all__ = ['resnest50', 'resnest101', 'resnest200', 'resnest269']
 from .build import RESNEST_MODELS_REGISTRY
 
-_url_format = 'https://s3.us-west-1.wasabisys.com/resnest/torch/{}-{}.pth'
+_url_format = 'https://github.com/zhanghang1989/ResNeSt/releases/download/weights_step1/{}-{}.pth'
 
 _model_sha256 = {name: checksum for checksum, name in [
     ('528c19ca', 'resnest50'),