bug fix for the latest torchvision

c728193f · Kai Chen · e2e109e4 · c728193f
Commit c728193f authored May 09, 2019 by Kai Chen
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 2 deletions

mmcv/runner/checkpoint.py mmcv/runner/checkpoint.py +3 -2

No files found.
--- a/mmcv/runner/checkpoint.py
+++ b/mmcv/runner/checkpoint.py
@@ -107,8 +107,9 @@ def load_checkpoint(model,
                torchvision.models.__path__):
            if not ispkg:
                _zoo = import_module('torchvision.models.{}'.format(name))
-                _urls = getattr(_zoo, 'model_urls')
-                model_urls.update(_urls)
+                if hasattr(_zoo, 'model_urls'):
+                    _urls = getattr(_zoo, 'model_urls')
+                    model_urls.update(_urls)
        model_name = filename[11:]
        checkpoint = model_zoo.load_url(model_urls[model_name])
    elif filename.startswith('open-mmlab://'):