We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 19069c4 commit 2275b35Copy full SHA for 2275b35
pytorch_lightning/core/decorators.py
@@ -89,7 +89,7 @@ def parameter_validation(fn: Callable) -> Callable:
89
def inner_f(self, *args, **kwargs):
90
pre_param_count = len(list(self.parameters()))
91
module = fn(self, *args, **kwargs)
92
- self.on_post_move_to_device()
+ self.trainer.call_hook('on_post_move_to_device')
93
post_param_count = len(list(self.parameters()))
94
95
if not pre_param_count == post_param_count:
0 commit comments