Skip to content

Commit

Permalink
call hook on ttp instead of acc
Browse files Browse the repository at this point in the history
  • Loading branch information
awaelchli committed Dec 6, 2021
1 parent e0de129 commit dd30690
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion pytorch_lightning/trainer/trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -1538,7 +1538,7 @@ def _call_accelerator_hook(
**kwargs: Any,
) -> Optional[Any]:
self.lightning_module._current_fx_name = hook_name
fn = getattr(self.accelerator, hook_name)
fn = getattr(self.training_type_plugin, hook_name)
if not callable(fn):
return None

Expand Down

0 comments on commit dd30690

Please sign in to comment.