Skip to content

GitLab

Commit 50f2ad21 authored by Kai Wang (Victor Kai)'s avatar Kai Wang (Victor Kai) Committed by Frank Lee
Browse files

[NFC] polish colossalai/engine/ophooks/utils.py code style (#1256)

parent 2dd4d556
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
colossalai/engine/ophooks/utils.py
View file @ 50f2ad21
...@@ -93,7 +93,7 @@ def register_ophooks_recursively(module: torch.nn.Module, ...@@ -93,7 +93,7 @@ def register_ophooks_recursively(module: torch.nn.Module,
assert isinstance(ophook_list, (list, tuple)) assert isinstance(ophook_list, (list, tuple))
assert len(ophook_list) > 0, 'expected at least 1 hook in the argument ophook_list but found 0' assert len(ophook_list) > 0, 'expected at least 1 hook in the argument ophook_list but found 0'
for hook in ophook_list: for hook in ophook_list:
assert (isinstance(hook, BaseOpHook)) assert (isinstance(hook, BaseOpHook))
# Add hooks for submodules # Add hooks for submodules
for child_name, child in module.named_children(): for child_name, child in module.named_children():
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment