From 622dba0168014a43363359312689157e5bb94e33 Mon Sep 17 00:00:00 2001 From: Philipp Horstenkamp Date: Wed, 26 Oct 2022 18:11:17 +0200 Subject: [PATCH] Fixes #2774 by updating the overall_progress_tqdm to full when stopping the progress bar now disappears. --- tensorflow_addons/callbacks/tqdm_progress_bar.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tensorflow_addons/callbacks/tqdm_progress_bar.py b/tensorflow_addons/callbacks/tqdm_progress_bar.py index 46459b2b62..c17630f89a 100644 --- a/tensorflow_addons/callbacks/tqdm_progress_bar.py +++ b/tensorflow_addons/callbacks/tqdm_progress_bar.py @@ -143,6 +143,9 @@ def _initialize_progbar(self, hook, epoch, logs=None): def _clean_up_progbar(self, hook, logs): if hook == "train_overall": if self.show_overall_progress: + self.overall_progress_tqdm.update( + self.num_epochs - self.overall_progress_tqdm.n + ) self.overall_progress_tqdm.close() else: if hook == "test":