diff --git a/pytorch_lightning/core/lightning.py b/pytorch_lightning/core/lightning.py index 5acec2b867..f66055bcca 100644 --- a/pytorch_lightning/core/lightning.py +++ b/pytorch_lightning/core/lightning.py @@ -1397,7 +1397,7 @@ class LightningModule( if running_train_loss is not None else float("NaN") ) - tqdm_dict = {"loss": "{:.3f}".format(avg_training_loss)} + tqdm_dict = {"loss": "{:.3g}".format(avg_training_loss)} if self.trainer.truncated_bptt_steps is not None: tqdm_dict["split_idx"] = self.trainer.split_idx