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

This commit is contained in:
Harsh Sharma authored and William Falcon committed 2020-01-21 16:48:30 -05:00
1 parent eeb48ceb96
commit 432a0bcd06
1 file changed
+1
@@ -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()