From ef72c3cd470dd67836eebf95ec567199c890a6a2 Mon Sep 17 00:00:00 2001
From: LittleNyima <littlenyima@163.com>
Date: Wed, 8 Nov 2023 18:15:18 +0800
Subject: [PATCH] Add warnings for duplicate registry names

---
 timm/models/_registry.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/timm/models/_registry.py b/timm/models/_registry.py
index 89c5c30c..a129e1af 100644
--- a/timm/models/_registry.py
+++ b/timm/models/_registry.py
@@ -86,6 +86,12 @@ def register_model(fn: Callable[..., Any]) -> Callable[..., Any]:
         mod.__all__ = [model_name]  # type: ignore
 
     # 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_to_module[model_name] = module_name
     _module_to_models[module_name].add(model_name)