early epoch stopping

2019-04-23 08:27:27 -04:00 · 2019-04-23 08:27:27 -04:00 · ffd6dc678c
parent 1961a6abb2
commit ffd6dc678c
1 changed files with 1 additions and 1 deletions
--- 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'):