Skip to content

Commit 2275b35

Browse files
committed
fix callbacks error
1 parent 19069c4 commit 2275b35

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pytorch_lightning/core/decorators.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -89,7 +89,7 @@ def parameter_validation(fn: Callable) -> Callable:
8989
def inner_f(self, *args, **kwargs):
9090
pre_param_count = len(list(self.parameters()))
9191
module = fn(self, *args, **kwargs)
92-
self.on_post_move_to_device()
92+
self.trainer.call_hook('on_post_move_to_device')
9393
post_param_count = len(list(self.parameters()))
9494

9595
if not pre_param_count == post_param_count:

0 commit comments

Comments
 (0)