Commit c728193f authored by Kai Chen's avatar Kai Chen
Browse files

bug fix for the latest torchvision

parent e2e109e4
......@@ -107,6 +107,7 @@ def load_checkpoint(model,
torchvision.models.__path__):
if not ispkg:
_zoo = import_module('torchvision.models.{}'.format(name))
if hasattr(_zoo, 'model_urls'):
_urls = getattr(_zoo, 'model_urls')
model_urls.update(_urls)
model_name = filename[11:]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment