diff --git a/pytorch_lightning/trainer/evaluation_loop.py b/pytorch_lightning/trainer/evaluation_loop.py index 6df04952..b5e2fe95 100644 --- a/pytorch_lightning/trainer/evaluation_loop.py +++ b/pytorch_lightning/trainer/evaluation_loop.py @@ -294,8 +294,8 @@ class TrainerEvaluationLoopMixin(ABC): position = 2 * self.process_position + (not test) desc = 'Testing' if test else 'Validating' pbar = tqdm(desc=desc, total=max_batches, leave=test, position=position, - disable=not self.show_progress_bar, dynamic_ncols=True, - unit='batch', file=sys.stdout) + disable=not self.show_progress_bar, dynamic_ncols=True, + unit='batch', file=sys.stdout) setattr(self, f'{"test" if test else "val"}_progress_bar', pbar) # run evaluation diff --git a/pytorch_lightning/trainer/trainer.py b/pytorch_lightning/trainer/trainer.py index 5466bc9d..9441f63e 100644 --- a/pytorch_lightning/trainer/trainer.py +++ b/pytorch_lightning/trainer/trainer.py @@ -829,8 +829,8 @@ class Trainer(TrainerIOMixin, # init progress bar pbar = tqdm(leave=True, position=2 * self.process_position, - disable=not self.show_progress_bar, dynamic_ncols=True, unit='batch', - file=sys.stdout) + disable=not self.show_progress_bar, dynamic_ncols=True, unit='batch', + file=sys.stdout) self.main_progress_bar = pbar # clear cache before training