Updated sig figs

NCAR · Feb 6, 2025 · 2436325 · 2436325
1 parent 8304d55
commit 2436325
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/credit/trainers/trainerERA5_multistep_grad_accum.py b/credit/trainers/trainerERA5_multistep_grad_accum.py
@@ -391,7 +391,7 @@ def train_one_epoch(
             )
             ensemble_size = conf["trainer"].get("ensemble_size", 0)
             if ensemble_size > 1:
-                to_print += f" std: {np.mean(results_dict['train_std'])}"
+                to_print += f" std: {np.mean(results_dict['train_std']):.6f}"
             to_print += " lr: {:.12f}".format(optimizer.param_groups[0]["lr"])
             if self.rank == 0:
                 batch_group_generator.update(1)