mirrors/mmengine
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmengine.git synced 2025-06-03 21:54:44 +08:00
Code Issues Projects Releases Wiki Activity

[Fix] Fix error implementation of is_model_wrapper (#640)

Browse Source 
* [Fix] Fix is_model_wrapper

* [Fix] Fix is_model_wrapper
This commit is contained in:
Mashiro 2022-10-28 15:16:56 +08:00 committed by GitHub
parent 0857f9fb40
commit 1bf5c0c12e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mmengine/model/wrappers/utils.py
View File

@ -26,5 +26,5 @@ def is_model_wrapper(model: nn.Module, registry: Registry = MODEL_WRAPPERS):
if not registry.children:
return False
for child in registry.children.values():
return is_model_wrapper(model, child)
return any(
is_model_wrapper(model, child) for child in registry.children.values())