Skip to content
Snippets Groups Projects
Commit ac77e3bf authored by duzx16's avatar duzx16
Browse files

Fix typo in collect_hooks_

parent 51e1a513
No related branches found
No related tags found
No related merge requests found
......@@ -115,7 +115,7 @@ class BaseModel(torch.nn.Module):
for mixin_name, m in self.mixins.items():
if hasattr(m, name):
if name in hooks: # if this hook name is already registered
if hasattr(getattr(m, name), 'non_confict'):
if hasattr(getattr(m, name), 'non_conflict'):
hooks[name] = partial(getattr(m, name), old_impl=hooks[name])
hook_origins[name] = mixin_name + ' -> ' + hook_origins[name]
else: # conflict
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment