logging average metrics

2023-03-09 17:59:18 +01:00 · 2023-03-09 17:59:18 +01:00 · 5ef0904e0e
parent 7e1ec46ebd
commit 5ef0904e0e
1 changed files with 5 additions and 1 deletions
--- a/gfun/vgfs/commons.py
+++ b/gfun/vgfs/commons.py
@ -218,7 +218,11 @@ class Trainer:
                    )

                wandb_logger.log(
-                    {"loss/val": eval_loss, **format_langkey_wandb(lang_metrics)},
+                    {
+                        "loss/val": eval_loss,
+                        **format_langkey_wandb(lang_metrics),
+                        "average/metrics": avg_metrics,
+                    },
                    commit=False,
                )