From e0a5aee3a3bbaa2e64d71cf6788187b57f743b25 Mon Sep 17 00:00:00 2001 From: Ashraful Islam Date: Mon, 18 May 2020 20:33:51 -0400 Subject: [PATCH] fix porgressbar postfix order (#1874) --- pytorch_lightning/callbacks/progress.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pytorch_lightning/callbacks/progress.py b/pytorch_lightning/callbacks/progress.py index fa7d0d7d..5d2c6f74 100644 --- a/pytorch_lightning/callbacks/progress.py +++ b/pytorch_lightning/callbacks/progress.py @@ -323,7 +323,7 @@ class ProgressBar(ProgressBarBase): super().on_batch_end(trainer, pl_module) if self.is_enabled and self.train_batch_idx % self.refresh_rate == 0: self.main_progress_bar.update(self.refresh_rate) - self.main_progress_bar.set_postfix(**trainer.progress_bar_dict) + self.main_progress_bar.set_postfix(trainer.progress_bar_dict) def on_validation_start(self, trainer, pl_module): super().on_validation_start(trainer, pl_module) @@ -338,7 +338,7 @@ class ProgressBar(ProgressBarBase): def on_validation_end(self, trainer, pl_module): super().on_validation_end(trainer, pl_module) - self.main_progress_bar.set_postfix(**trainer.progress_bar_dict) + self.main_progress_bar.set_postfix(trainer.progress_bar_dict) self.val_progress_bar.close() def on_train_end(self, trainer, pl_module):