diff --git a/SwissArmyTransformer/model/base_model.py b/SwissArmyTransformer/model/base_model.py index 61543f5d30ed922921b6cd9804b08550e894ca00..50fe20293c87ccf2a32da8e82c2dc880dbd4580f 100644 --- a/SwissArmyTransformer/model/base_model.py +++ b/SwissArmyTransformer/model/base_model.py @@ -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