Skip to content

Commit

Permalink
Fix wrong name in _all_ for models._registry
Browse files Browse the repository at this point in the history
  • Loading branch information
rwightman committed Oct 15, 2024
1 parent b4a9a16 commit 60f517c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion timm/models/_registry.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 60f517c

Please sign in to comment.