Issue #657 - Call on_train_end after early stopping (#723)

2020-01-21 14:48:31 -07:00 · 2020-01-21 14:48:31 -07:00 · 432a0bcd06
parent eeb48ceb96
commit 432a0bcd06
1 changed files with 1 additions and 0 deletions
--- a/pytorch_lightning/trainer/training_loop.py
+++ b/pytorch_lightning/trainer/training_loop.py
@ -353,6 +353,7 @@ class TrainerTrainLoopMixin(ABC):
                stop = should_stop and met_min_epochs
                if stop:
                    self.main_progress_bar.close()
+                    model.on_train_end()
                    return

        self.main_progress_bar.close()