propagating eval_batch_size to TrainingArguments (huggingface#675)

Co-authored-by: Rahul Jha <rahuljha@netflix.com>
kushalarora · Sep 19, 2023 · eb2ebe0 · eb2ebe0
1 parent a6d3141
commit eb2ebe0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/examples/research_projects/stack_llama_2/scripts/dpo_llama2.py b/examples/research_projects/stack_llama_2/scripts/dpo_llama2.py
@@ -164,6 +164,7 @@ def return_prompt_and_responses(samples) -> Dict[str, str]:
     # 4. initialize training arguments:
     training_args = TrainingArguments(
         per_device_train_batch_size=script_args.per_device_train_batch_size,
+        per_device_eval_batch_size=script_args.per_device_eval_batch_size,
         max_steps=script_args.max_steps,
         logging_steps=script_args.logging_steps,
         save_steps=script_args.save_steps,