mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-27 13:12:20 +08:00
fixed hooks
This commit is contained in:
1 parent
4b0b7e5ea3
commit
333f0fde9b
1 file changed
+1
-1
@@ -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
|
||||
|
||||
Reference in new issue
Block a user