From 25bb3864e87fffef384569ee7721aa24277dc4db Mon Sep 17 00:00:00 2001 From: 51616 Date: Wed, 25 Dec 2024 08:50:43 +0000 Subject: [PATCH] fix rouge metric prefix --- hyperlora/training_utils.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/hyperlora/training_utils.py b/hyperlora/training_utils.py index 137100b..2b8458c 100644 --- a/hyperlora/training_utils.py +++ b/hyperlora/training_utils.py @@ -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)