andrea.pedrotti
/
gfun_multimodal
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

logging average metrics

This commit is contained in:
andreapdr 2023-03-09 17:59:18 +01:00
parent 7e1ec46ebd
commit 5ef0904e0e
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
gfun/vgfs/commons.py
View File

@ -218,7 +218,11 @@ class Trainer:
) )
wandb_logger.log( 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, commit=False,
) )