Skip to content

Commit 43e20b3

Browse files
yuisekiyuiseki
committed
update recipes/A5000_24GB_x8/sentiment-analyzer-en.yaml
1 parent 140429f commit 43e20b3

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

‎recipes/A5000_24GB_x8/sentiment-analyzer-en.yaml

+2-2
Original file line numberDiff line numberDiff line change
@@ -14,8 +14,8 @@ lora_r: 8
1414
lora_alpha: 16
1515
lora_dropout: 0.05
1616
train_claim_gpu_num: 1
17-
train_per_device_train_batch_size: 16
18-
train_gradient_accumulation_steps: 4
17+
train_per_device_train_batch_size: 8
18+
train_gradient_accumulation_steps: 2
1919
train_num_train_epochs: 4
2020
train_max_steps: 1000
2121
train_fp16: True

0 commit comments

Comments
 (0)