chore: change lr to original
This commit is contained in:
parent
0dbc298095
commit
e6e5cd97e1
|
@ -25,7 +25,7 @@ overwrite_output_dir: true
|
||||||
### train
|
### train
|
||||||
per_device_train_batch_size: 2
|
per_device_train_batch_size: 2
|
||||||
gradient_accumulation_steps: 8
|
gradient_accumulation_steps: 8
|
||||||
learning_rate: 1.0e-5
|
learning_rate: 1.0e-4
|
||||||
num_train_epochs: 10.0
|
num_train_epochs: 10.0
|
||||||
lr_scheduler_type: cosine
|
lr_scheduler_type: cosine
|
||||||
warmup_ratio: 0.1
|
warmup_ratio: 0.1
|
||||||
|
|
|
@ -3,6 +3,7 @@
|
||||||
|
|
||||||
# bash run_once.sh lora_sft Qwen-7B 8 50
|
# bash run_once.sh lora_sft Qwen-7B 8 50
|
||||||
# bash run_once.sh lora_sft Qwen-7B 1 50
|
# bash run_once.sh lora_sft Qwen-7B 1 50
|
||||||
|
# bash run_once.sh lora_sft 9g-8B 8 50
|
||||||
# bash run_once.sh inference Qwen-7B 8 50
|
# bash run_once.sh inference Qwen-7B 8 50
|
||||||
run_type="$1"
|
run_type="$1"
|
||||||
model="$2"
|
model="$2"
|
||||||
|
|
Loading…
Reference in New Issue