Skip to content

Commit 432a0bc

Browse files
HarshSharma12williamFalcon
authored andcommitted
Issue #657 - Call on_train_end after early stopping (#723)
1 parent eeb48ce commit 432a0bc

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

pytorch_lightning/trainer/training_loop.py

+1
Original file line numberDiff line numberDiff line change
@@ -353,6 +353,7 @@ def train(self):
353353
stop = should_stop and met_min_epochs
354354
if stop:
355355
self.main_progress_bar.close()
356+
model.on_train_end()
356357
return
357358

358359
self.main_progress_bar.close()

0 commit comments

Comments
 (0)