Update resnet.py

yassouali · Jun 30, 2020 · 406d8ad · 406d8ad
1 parent f19cdcb
commit 406d8ad
Showing 1 changed file with 4 additions and 4 deletions.
diff --git 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)
+    return torch.load(cached_file, map_location=map_location)