Skip to content

GitLab

"src/vscode:/vscode.git/clone" did not exist on "2d35f6733a2d698e8917896071444a5923993ae7"
Commit c728193f authored by Kai Chen's avatar Kai Chen
Browse files

bug fix for the latest torchvision

parent e2e109e4
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
mmcv/runner/checkpoint.py
View file @ c728193f
...@@ -107,6 +107,7 @@ def load_checkpoint(model, ...@@ -107,6 +107,7 @@ def load_checkpoint(model,
torchvision.models.__path__): torchvision.models.__path__):
if not ispkg: if not ispkg:
_zoo = import_module('torchvision.models.{}'.format(name)) _zoo = import_module('torchvision.models.{}'.format(name))
if hasattr(_zoo, 'model_urls'):
_urls = getattr(_zoo, 'model_urls') _urls = getattr(_zoo, 'model_urls')
model_urls.update(_urls) model_urls.update(_urls)
model_name = filename[11:] 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