From 333f0fde9b6e5c8cceeb8d1238df05f66c6855ae Mon Sep 17 00:00:00 2001 From: William Falcon Date: Sun, 21 Apr 2019 14:16:54 -0400 Subject: [PATCH] fixed hooks --- pytorch_lightning/models/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index 2a23cf62..2b1114a6 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -86,7 +86,7 @@ class Trainer(TrainerIO): self.test_percent_check = overfit_pct def __is_function_implemented(self, f_name): - f_op = getattr(self, f_name, None) + f_op = getattr(self.model, f_name, None) return callable(f_op) @property