Skip to content

Commit 1bf5c0c

Browse files
authored
[Fix] Fix error implementation of is_model_wrapper (#640)
* [Fix] Fix is_model_wrapper * [Fix] Fix is_model_wrapper
1 parent 0857f9f commit 1bf5c0c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

mmengine/model/wrappers/utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,5 +26,5 @@ def is_model_wrapper(model: nn.Module, registry: Registry = MODEL_WRAPPERS):
2626
if not registry.children:
2727
return False
2828

29-
for child in registry.children.values():
30-
return is_model_wrapper(model, child)
29+
return any(
30+
is_model_wrapper(model, child) for child in registry.children.values())

0 commit comments

Comments
 (0)