Add eval_batch_size for evaluation

This commit is contained in:
NanoCode012
2023-05-06 22:21:24 +09:00
parent c0f50d9c61
commit 0e74b6402e
2 changed files with 2 additions and 0 deletions

View File

@@ -47,6 +47,7 @@ def setup_trainer(cfg, train_dataset, eval_dataset, model, tokenizer):
training_args = transformers.TrainingArguments(
per_device_train_batch_size=cfg.micro_batch_size,
per_device_eval_batch_size=cfg.eval_batch_size,
gradient_accumulation_steps=cfg.gradient_accumulation_steps,
num_train_epochs=cfg.num_epochs,
learning_rate=cfg.learning_rate,