From 7ec4ef66bb75a8385457dc68ea04e35e808a7f26 Mon Sep 17 00:00:00 2001 From: hiyyg <62354758+hiyyg@users.noreply.github.com> Date: Fri, 18 Dec 2020 14:58:10 +0800 Subject: [PATCH] download pretrained models with progress --- timm/models/helpers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/timm/models/helpers.py b/timm/models/helpers.py index 77b98dc6..3d785af1 100644 --- 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)