[fix] evaluation dataset is incorrect in reward-model/trainer.py

2026-06-27 16:10:30 +08:00 · 2023-01-08 03:42:40 +00:00
parent 9d05b73efc
commit a1e1445de9
1 changed files with 1 additions and 1 deletions
@@ -182,7 +182,7 @@ if __name__ == "__main__":
        args=args,
        loss_function=training_conf["loss"],
        train_dataset=train,
-        eval_dataset=eval,
+        eval_dataset=evals,
        data_collator=collate_fn,
        tokenizer=tokenizer,
        compute_metrics=compute_metrics,