Skip to content

Commit

Permalink
Update resnet.py
Browse files Browse the repository at this point in the history
  • Loading branch information
yassouali authored Jun 30, 2020
1 parent f19cdcb commit 406d8ad
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions models/resnet.py
Original file line number Diff line number Diff line change
Expand Up @@ -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'
}


Expand Down Expand Up @@ -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)

0 comments on commit 406d8ad

Please sign in to comment.