From bb3a8071993f1a0662cda52f0eb8278b9dfe1780 Mon Sep 17 00:00:00 2001 From: Jintao Huang Date: Fri, 21 Feb 2025 18:12:57 +0800 Subject: [PATCH] update --- swift/trainers/rlhf_trainer/grpo_trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/swift/trainers/rlhf_trainer/grpo_trainer.py b/swift/trainers/rlhf_trainer/grpo_trainer.py index 0e2406bfb..e26c50da8 100644 --- a/swift/trainers/rlhf_trainer/grpo_trainer.py +++ b/swift/trainers/rlhf_trainer/grpo_trainer.py @@ -522,6 +522,6 @@ def _get_per_token_logps(self, model, inputs): def evaluation_loop(self, *args, **kwargs): metric_key_prefix = kwargs['metric_key_prefix'] output = super().evaluation_loop(*args, **kwargs) - metrics = {f'{metric_key_prefix}_{key}': sum(val) / len(val) for key, val in self._metrics.items()} + metrics = {f'{metric_key_prefix}_{key}': sum(val) / len(val) for key, val in self._metrics['eval'].items()} output.metrics.update(metrics) return output