From c16fec9252f146fae0380b8c386eb1502548c646 Mon Sep 17 00:00:00 2001 From: Daniel Nouri Date: Wed, 6 Apr 2022 10:24:12 +0200 Subject: [PATCH 1/2] In load_pretrained helper: Fail if cannot find pretrained model Previously, this would only log a warning. --- 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 c4f48d6a..aa2ea370 100644 --- a/timm/models/helpers.py +++ b/timm/models/helpers.py @@ -248,7 +248,7 @@ def load_pretrained( _logger.info(f'Loading pretrained weights from Hugging Face hub ({pretrained_loc})') state_dict = load_state_dict_from_hf(pretrained_loc) else: - _logger.warning("No pretrained weights exist or were found for this model. Using random initialization.") + raise ValueError("No pretrained weights exist or were found for this model.") return if filter_fn is not None: From 2be436cb2cdf6afb25acb861b4def34bf21fa801 Mon Sep 17 00:00:00 2001 From: Daniel Nouri Date: Thu, 7 Apr 2022 08:52:17 +0200 Subject: [PATCH 2/2] Remove dead code return Co-authored-by: Hyeongchan Kim --- timm/models/helpers.py | 1 - 1 file changed, 1 deletion(-) diff --git a/timm/models/helpers.py b/timm/models/helpers.py index aa2ea370..67244d28 100644 --- a/timm/models/helpers.py +++ b/timm/models/helpers.py @@ -249,7 +249,6 @@ def load_pretrained( state_dict = load_state_dict_from_hf(pretrained_loc) else: raise ValueError("No pretrained weights exist or were found for this model.") - return if filter_fn is not None: # for backwards compat with filter fn that take one arg, try one first, the two