Loss format from .3f to .3g in the tqdm (Lightning-AI#4972)

Co-authored-by: Jirka Borovec <[email protected]>
gianscarpe · Dec 9, 2020 · 068502f · 068502f
1 parent bcbba3b
commit 068502f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_lightning/core/lightning.py b/pytorch_lightning/core/lightning.py
@@ -1397,7 +1397,7 @@ def get_progress_bar_dict(self):
             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