diff --git a/gfun/vgfs/commons.py b/gfun/vgfs/commons.py index 6399d85..9a317af 100644 --- 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, )