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

2026-09-10 12:21:57 +08:00 · 2020-01-21 16:48:30 -05:00
parent eeb48ceb96
commit 432a0bcd06
1 changed files with 1 additions and 0 deletions
@@ -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()