Spaces:

vector2000
/

test_finetunning

Paused

vector2000 commited on Aug 27, 2024

Commit

6e5f17b

verified ·

1 Parent(s): eaf4aec

Update finetunning.py

Files changed (1) hide show

finetunning.py CHANGED Viewed

@@ -54,7 +54,7 @@ training_args = TrainingArguments(
     output_dir="./results",
     overwrite_output_dir=True,
     num_train_epochs=3,
-    per_device_train_batch_size=4,
     save_steps=10_000,
     save_total_limit=2,
     prediction_loss_only=True,

     output_dir="./results",
     overwrite_output_dir=True,
     num_train_epochs=3,
+    per_device_train_batch_size=32,
     save_steps=10_000,
     save_total_limit=2,
     prediction_loss_only=True,