From a1e1445de9c73d57914f5ac9280f608ddfbb104c Mon Sep 17 00:00:00 2001 From: theblackcat102 Date: Sun, 8 Jan 2023 03:42:40 +0000 Subject: [PATCH] [fix] evaluation dataset is incorrect in reward-model/trainer.py --- model/reward/instructor/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model/reward/instructor/trainer.py b/model/reward/instructor/trainer.py index 7500901b..940c0708 100644 --- a/model/reward/instructor/trainer.py +++ b/model/reward/instructor/trainer.py @@ -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,