diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index 1bac00a63b..8039d03870 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -326,7 +326,7 @@ class Trainer(TrainerIO): def __run_tng_batch(self, data_batch): if data_batch is None: - return + return 0 # hook if self.__is_function_implemented('on_batch_start'):