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

fjhheras · Borda · web-flow · commit 068502f07cfd · 2020-12-09T14:13:13.000-05:00
Co-authored-by: Jirka Borovec &lt;Borda@users.noreply.github.com&gt;
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

Original file line number	Diff line number	Diff line change
`@@ -1397,7 +1397,7 @@ def get_progress_bar_dict(self):`
`1397`	`1397`	`if running_train_loss is not None`
`1398`	`1398`	`else float("NaN")`
`1399`	`1399`	`)`
`1400`		`- tqdm_dict = {"loss": "{:.3f}".format(avg_training_loss)}`
	`1400`	`+ tqdm_dict = {"loss": "{:.3g}".format(avg_training_loss)}`
`1401`	`1401`
`1402`	`1402`	`if self.trainer.truncated_bptt_steps is not None:`
`1403`	`1403`	`tqdm_dict["split_idx"] = self.trainer.split_idx`