[shardformer] added embedding gradient check (#4124)

2025-09-07 12:01:39 +00:00 · 2023-06-30 16:16:44 +08:00
parent 44a190e6ac
commit ae035d305d
14 changed files with 255 additions and 74 deletions
--- a/tests/kit/model_zoo/registry.py
+++ b/tests/kit/model_zoo/registry.py
@@ -70,6 +70,8 @@ class ModelZooRegistry(dict):
        for k, v in self.items():
            if keyword in k:
                new_dict[k] = v
+
+        assert len(new_dict) > 0, f'No model found with keyword {keyword}'
        return new_dict