From c1ecca418e762af000bb63ddcc72c00dcee9ad3b Mon Sep 17 00:00:00 2001 From: Tullie Murrell Date: Thu, 21 Nov 2019 10:26:24 -0800 Subject: [PATCH] Write progress bar to stdout (#531) * Default write progress bar to stdout * Change validation progress too --- pytorch_lightning/trainer/evaluation_loop_mixin.py | 3 ++- pytorch_lightning/trainer/trainer.py | 4 +++- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/pytorch_lightning/trainer/evaluation_loop_mixin.py b/pytorch_lightning/trainer/evaluation_loop_mixin.py index 8583805ab7..0b7a132471 100644 --- a/pytorch_lightning/trainer/evaluation_loop_mixin.py +++ b/pytorch_lightning/trainer/evaluation_loop_mixin.py @@ -1,4 +1,5 @@ import torch +import sys import tqdm from pytorch_lightning.utilities.debugging import MisconfigurationException @@ -120,7 +121,7 @@ class TrainerEvaluationLoopMixin(object): desc = 'Testing' if test else 'Validating' pbar = tqdm.tqdm(desc=desc, total=max_batches, leave=test, position=position, disable=not self.show_progress_bar, dynamic_ncols=True, - unit='batch') + 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 eebba265c3..155a6a7fdb 100644 --- a/pytorch_lightning/trainer/trainer.py +++ b/pytorch_lightning/trainer/trainer.py @@ -3,6 +3,7 @@ The trainer handles all the logic for running a val loop, training loop, distrib """ import os +import sys import warnings import logging @@ -460,7 +461,8 @@ class Trainer(TrainerIOMixin, # init progress bar pbar = tqdm.tqdm(leave=True, position=2 * self.process_position, - disable=not self.show_progress_bar, dynamic_ncols=True, unit='batch') + disable=not self.show_progress_bar, dynamic_ncols=True, unit='batch', + file=sys.stdout) self.main_progress_bar = pbar # clear cache before training