Set per_device_batch_sizes = 3

arguments.py

@@ -9,8 +9,8 @@ from transformers import TrainingArguments
 class HFTrainerArguments(TrainingArguments):
     """Arguments for huggingface/transformers.Trainer"""
     dataloader_num_workers: int = 1
-    per_device_train_batch_size: int = 2
-    per_device_eval_batch_size: int = 2
+    per_device_train_batch_size: int = 3
+    per_device_eval_batch_size: int = 3
     gradient_accumulation_steps: int = 1
     text_seq_length: int = 256