From 3f543f1137796de3e665ffa5402358080eb7b169 Mon Sep 17 00:00:00 2001 From: Emmanuel Ferdman Date: Mon, 9 Jun 2025 16:20:18 -0700 Subject: [PATCH] Throw exception on non-directory path for pretrained weights Signed-off-by: Emmanuel Ferdman --- timm/models/_builder.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/timm/models/_builder.py b/timm/models/_builder.py index 2b34ead6b4..03e5dc3a8f 100644 --- a/timm/models/_builder.py +++ b/timm/models/_builder.py @@ -220,7 +220,7 @@ def load_pretrained( if pretrained_path.is_dir(): state_dict = load_state_dict_from_path(pretrained_path) else: - RuntimeError(f"Specified path is not a directory: {pretrained_loc}") + raise RuntimeError(f"Specified path is not a directory: {pretrained_loc}") else: model_name = pretrained_cfg.get('architecture', 'this model') raise RuntimeError(f"No pretrained weights exist for {model_name}. Use `pretrained=False` for random init.")