apunkt/doc-to-lora
1
0
Fork 0
mirror of https://github.com/SakanaAI/doc-to-lora.git synced 2026-07-23 17:01:04 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix rouge metric prefix

Browse source
This commit is contained in:
51616 2024-12-25 08:50:43 +00:00
parent b555f1d07a
commit 25bb3864e8
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
hyperlora/training_utils.py
View file

@ -72,7 +72,8 @@ def eval_generation(eval_trainer, tokenizer, dataset, split, gen_kwargs):
[txt["generated"] for txt in decoded_txts],
[txt["label"] for txt in decoded_txts],
)
eval_result.metrics.update(rouge_metrics)
for k, v in rouge_metrics.items():
eval_result.metrics[f"{split}_{k}"] = v
eval_trainer.log_metrics("eval" if split == "val" else split, eval_result.metrics)
eval_trainer.save_metrics("eval" if split == "val" else split, eval_result.metrics)