mirrors
/
pytorch-image-models
mirror of https://github.com/huggingface/pytorch-image-models.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Add warnings for duplicate registry names

pull/2029/head
LittleNyima 2023-11-08 18:15:18 +08:00 committed by Ross Wightman
parent 0bc9a63ee5
commit ef72c3cd47
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
timm/models/_registry.py
View File

@ -86,6 +86,12 @@ def register_model(fn: Callable[..., Any]) -> Callable[..., Any]:
mod.__all__ = [model_name] # type: ignore mod.__all__ = [model_name] # type: ignore
# add entries to registry dict/sets # add entries to registry dict/sets
if model_name in _model_entrypoints:
warnings.warn(
f'Overwriting {model_name} in registry with {fn.__module__}.{model_name}. This is because the name being '
'registered conflicts with an existing name. Please check if this is not expected.',
stacklevel=2,
)
_model_entrypoints[model_name] = fn _model_entrypoints[model_name] = fn
_model_to_module[model_name] = module_name _model_to_module[model_name] = module_name
_module_to_models[module_name].add(model_name) _module_to_models[module_name].add(model_name)