batch size 2

scripts/train.py

@@ -70,7 +70,7 @@ def main():
     # Training arguments
     training_args = TrainingArguments(
         output_dir="./output",
-        per_device_train_batch_size=4,
+        per_device_train_batch_size=2,
         gradient_accumulation_steps=8,
         learning_rate=1e-4,
         num_train_epochs=1,