From 60f517c88327a64d44fb7259bb427c7658fdbb1b Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Tue, 15 Oct 2024 07:39:46 -0700 Subject: [PATCH] Fix wrong name in _all_ for models._registry --- timm/models/_registry.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/timm/models/_registry.py b/timm/models/_registry.py index 09d73b551b..d8ec2f6439 100644 --- a/timm/models/_registry.py +++ b/timm/models/_registry.py @@ -16,7 +16,7 @@ __all__ = [ 'split_model_name_tag', 'get_arch_name', 'register_model', 'generate_default_cfgs', 'list_models', 'list_pretrained', 'is_model', 'model_entrypoint', 'list_modules', 'is_model_in_modules', - 'get_pretrained_cfg_value', 'is_model_pretrained', 'get_pretrained_cfgs_for_arch' + 'get_pretrained_cfg_value', 'is_model_pretrained', 'get_arch_pretrained_cfgs' ] _module_to_models: Dict[str, Set[str]] = defaultdict(set) # dict of sets to check membership of model in module