Skip to content

Commit bf34f04

Browse files
committed
update recipes/A5000_24GB_x8/sentiment-analyzer-en.yaml
1 parent 428dc82 commit bf34f04

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

recipes/A5000_24GB_x8/sentiment-analyzer-en.yaml

+9-3
Original file line numberDiff line numberDiff line change
@@ -14,10 +14,10 @@ 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: 6
18-
train_gradient_accumulation_steps: 2
17+
train_per_device_train_batch_size: 16
18+
train_gradient_accumulation_steps: 4
1919
train_num_train_epochs: 4
20-
train_max_steps: 200
20+
train_max_steps: 10000
2121
train_fp16: True
2222
inference_max_new_tokens: 16
2323
evaluations:
@@ -27,3 +27,9 @@ evaluations:
2727
-
2828
prompt: "Hmm don't know what to do today -- Model just canceled = I hate it when that happens..will have to do gardening! "
2929
expected_output: "Negative"
30+
-
31+
prompt: "About to go to bed. Sleeping really late tomorrow! I am so glad the Tigers won tonight!!"
32+
expected_output: "Positive"
33+
-
34+
prompt: "AHH i'm so HAPPY. I just found my ipod. God is sooo good to me!"
35+
expected_output: "Positive"

0 commit comments

Comments
 (0)