Merge branch 'trl0.16' of github.com:hjh0119/swift into trl0.16

hjh0119 · Feb 21, 2025 · 9e23cbd · 9e23cbd
2 parents 6d7cbc2 + 605a47a
commit 9e23cbd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/swift/trainers/rlhf_trainer/grpo_trainer.py b/swift/trainers/rlhf_trainer/grpo_trainer.py
@@ -521,6 +521,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