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()