From ed155c526871fb58be1f321cbaf23ccab32c7f19 Mon Sep 17 00:00:00 2001 From: Harsh Sharma Date: Tue, 21 Jan 2020 11:23:00 -0700 Subject: [PATCH] Issue #657 - Call on_train_end after early stopping --- 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 06d3ebd5c0b07..d5f16c9462bf1 100644 --- a/pytorch_lightning/trainer/training_loop.py +++ b/pytorch_lightning/trainer/training_loop.py @@ -353,6 +353,7 @@ def train(self): stop = should_stop and met_min_epochs if stop: self.main_progress_bar.close() + model.on_train_end() return self.main_progress_bar.close()