Update run.sh

hflyzju · Jul 21, 2023 · 21f17b8 · 21f17b8
1 parent 2bc76f9
commit 21f17b8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/run.sh b/run.sh
@@ -7,8 +7,8 @@ accelerate launch train.py \
     --learning_rate 3e-5 \
     --num_warmup_steps_or_radios 0.1 \
     --gradient_accumulation_steps 1 \
-    --per_device_train_batch_size 16 \
-    --per_device_eval_batch_size 32 \
+    --per_device_train_batch_size 4 \
+    --per_device_eval_batch_size 8 \
     --seed 42 \
     --save_steps 10804 \
     --output_dir ./outputs \