mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-28 13:20:31 +08: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()
|
||||
|
||||
Reference in new issue
Block a user