download pretrained models with progress

5 years ago · 7ec4ef66bb
parent 198f6ea0f3
commit 7ec4ef66bb
1 changed files with 1 additions and 1 deletions
--- a/timm/models/helpers.py
+++ b/timm/models/helpers.py
@ -95,7 +95,7 @@ def load_pretrained(model, cfg=None, num_classes=1000, in_chans=3, filter_fn=Non
        _logger.warning("Pretrained model URL is invalid, using random initialization.")
        return

-    state_dict = model_zoo.load_url(cfg['url'], progress=False, map_location='cpu')
+    state_dict = model_zoo.load_url(cfg['url'], map_location='cpu')

    if filter_fn is not None:
        state_dict = filter_fn(state_dict)