From 1dfb3e8867e2ece1c28a65c9db1cded2818a2031 Mon Sep 17 00:00:00 2001 From: Hang Zhang Date: Mon, 17 May 2021 17:52:29 -0700 Subject: [PATCH] Update ResNeSt Model Weights Step 1 (#156) * Update resnest.py * Update model_store.py * Update ablation.py --- resnest/gluon/model_store.py | 4 ++-- resnest/torch/models/ablation.py | 2 +- resnest/torch/models/resnest.py | 2 +- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/resnest/gluon/model_store.py b/resnest/gluon/model_store.py index fdc1826..effc3f5 100644 --- 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 index 00743cc..f91fe6a 100644 --- 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 index ed3594f..6308a21 100644 --- 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'),