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

Lightning-AI · Jan 21, 2020 · 432a0bc · 432a0bc
1 parent eeb48ce
commit 432a0bc
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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()