Update recipes/A5000_24GB_x8/translator-en-ja-math.yaml

yuiseki · Mar 21, 2024 · 84ee1f7 · 84ee1f7
1 parent 4d13953
commit 84ee1f7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/recipes/A5000_24GB_x8/translator-en-ja-math.yaml b/recipes/A5000_24GB_x8/translator-en-ja-math.yaml
@@ -12,7 +12,7 @@ lora_r: 8
 lora_alpha: 16
 lora_dropout: 0.05
 train_clain_gpu_num: 4
-train_per_device_train_batch_size: 8
+train_per_device_train_batch_size: 4
 train_gradient_accumulation_steps: 4
 train_num_train_epochs: 4
 train_max_steps: 1000