From 406d8ad2c99db9c768d8e752706ba72fc250f337 Mon Sep 17 00:00:00 2001 From: Yassine Date: Tue, 30 Jun 2020 12:06:31 +0200 Subject: [PATCH] Update resnet.py --- models/resnet.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/models/resnet.py b/models/resnet.py index c88777861..24ee0a4fa 100644 --- a/models/resnet.py +++ b/models/resnet.py @@ -21,9 +21,9 @@ model_urls = { 'resnet18': 'https://download.pytorch.org/models/resnet18-5c106cde.pth', 'resnet34': 'https://download.pytorch.org/models/resnet34-333f7ec4.pth', - 'resnet50': 'https://hangzh.s3.amazonaws.com/encoding/models/resnet50s-a75c83cf.zip', - 'resnet101': 'https://hangzh.s3.amazonaws.com/encoding/models/resnet101s-03a0f310.zip', - 'resnet152': 'https://hangzh.s3.amazonaws.com/encoding/models/resnet152s-36670e8b.zip' + 'resnet50': 'https://s3.us-west-1.wasabisys.com/encoding/models/resnet50s-a75c83cf.zip', + 'resnet101': 'https://s3.us-west-1.wasabisys.com/encoding/models/resnet101s-03a0f310.zip', + 'resnet152': 'https://s3.us-west-1.wasabisys.com/encoding/models/resnet152s-36670e8b.zip' } @@ -303,4 +303,4 @@ def load_url(url, model_dir='./pretrained', map_location=None): zip_ref.close() os.remove(cached_file) cached_file = os.path.join(model_dir, filename+'.pth') - return torch.load(cached_file, map_location=map_location) \ No newline at end of file + return torch.load(cached_file, map_location=map_location)