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,