From b41f76c7c502e92ba9bc586ad16e0a847a3e065d Mon Sep 17 00:00:00 2001 From: wassname Date: Sun, 26 Jan 2020 11:37:17 +0800 Subject: [PATCH] use `from tqdm.auto` in eval loop --- pytorch_lightning/trainer/evaluation_loop.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pytorch_lightning/trainer/evaluation_loop.py b/pytorch_lightning/trainer/evaluation_loop.py index e0148d60..6df04952 100644 --- a/pytorch_lightning/trainer/evaluation_loop.py +++ b/pytorch_lightning/trainer/evaluation_loop.py @@ -127,7 +127,7 @@ import sys from abc import ABC, abstractmethod import torch -import tqdm +from tqdm.auto import tqdm from pytorch_lightning.utilities.debugging import MisconfigurationException @@ -293,7 +293,7 @@ class TrainerEvaluationLoopMixin(ABC): # main progress bar will already be closed when testing so initial position is free position = 2 * self.process_position + (not test) desc = 'Testing' if test else 'Validating' - pbar = tqdm.tqdm(desc=desc, total=max_batches, leave=test, position=position, + 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) setattr(self, f'{"test" if test else "val"}_progress_bar', pbar)