From 3ad5809928f8a05121dc4010513cb6c62e6bf4b0 Mon Sep 17 00:00:00 2001 From: awaelchli Date: Fri, 8 Sep 2023 00:53:15 +0200 Subject: [PATCH] close validation progress bar before updating training bar --- src/lightning/pytorch/callbacks/progress/tqdm_progress.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/lightning/pytorch/callbacks/progress/tqdm_progress.py b/src/lightning/pytorch/callbacks/progress/tqdm_progress.py index fc36f81e8299b..98d8859e7d22d 100644 --- a/src/lightning/pytorch/callbacks/progress/tqdm_progress.py +++ b/src/lightning/pytorch/callbacks/progress/tqdm_progress.py @@ -240,8 +240,8 @@ def on_sanity_check_start(self, *_: Any) -> None: self.train_progress_bar = Tqdm(disable=True) # dummy progress bar def on_sanity_check_end(self, *_: Any) -> None: - self.train_progress_bar.close() self.val_progress_bar.close() + self.train_progress_bar.close() def on_train_start(self, *_: Any) -> None: self.train_progress_bar = self.init_train_tqdm() @@ -300,10 +300,10 @@ def on_validation_batch_end( _update_n(self.val_progress_bar, n) def on_validation_end(self, trainer: "pl.Trainer", pl_module: "pl.LightningModule") -> None: - if self._train_progress_bar is not None and trainer.state.fn == "fit": - self.train_progress_bar.set_postfix(self.get_metrics(trainer, pl_module)) self.val_progress_bar.close() self.reset_dataloader_idx_tracker() + if self._train_progress_bar is not None and trainer.state.fn == "fit": + self.train_progress_bar.set_postfix(self.get_metrics(trainer, pl_module)) def on_test_start(self, trainer: "pl.Trainer", pl_module: "pl.LightningModule") -> None: self.test_progress_bar = self.init_test_tqdm()