From 432a0bcd06b9d20e38b99fd279d27bcc58904ea0 Mon Sep 17 00:00:00 2001 From: Harsh Sharma Date: Tue, 21 Jan 2020 14:48:31 -0700 Subject: [PATCH] Issue #657 - Call on_train_end after early stopping (#723) --- pytorch_lightning/trainer/training_loop.py | 1 + 1 file changed, 1 insertion(+) diff --git a/pytorch_lightning/trainer/training_loop.py b/pytorch_lightning/trainer/training_loop.py index 06d3ebd5c0..d5f16c9462 100644 --- 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()