From c16fec9252f146fae0380b8c386eb1502548c646 Mon Sep 17 00:00:00 2001 From: Daniel Nouri Date: Wed, 6 Apr 2022 10:24:12 +0200 Subject: [PATCH] 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: