(self, *args, **kwargs)
| 131 | return self.mixins[name] |
| 132 | |
| 133 | def forward(self, *args, **kwargs): |
| 134 | # update hooks as the current model (overrided forwards) |
| 135 | # Attention! the transformer might be shared by multiple models |
| 136 | self.transformer.hooks.clear() |
| 137 | self.transformer.hooks.update(self.hooks) |
| 138 | return self.transformer(*args, **kwargs) |
| 139 | |
| 140 | def collect_hooks_(self): |
| 141 | names = list(HOOKS_DEFAULT.keys()) |