train: change yaml
This commit is contained in:
parent
2827dd1c98
commit
0841a0832f
|
@ -8,7 +8,7 @@ finetuning_type: lora
|
||||||
lora_target: q_proj,v_proj
|
lora_target: q_proj,v_proj
|
||||||
|
|
||||||
### dataset
|
### dataset
|
||||||
dataset: alpaca_gpt4_zh
|
dataset: alpaca_en
|
||||||
template: llama2
|
template: llama2
|
||||||
cutoff_len: 1024
|
cutoff_len: 1024
|
||||||
max_samples: 100000
|
max_samples: 100000
|
||||||
|
@ -16,7 +16,7 @@ overwrite_cache: true
|
||||||
preprocessing_num_workers: 16
|
preprocessing_num_workers: 16
|
||||||
|
|
||||||
### output
|
### output
|
||||||
output_dir: ./saves/LLaMA2-7B/lora/train_24_8_13_07_26
|
output_dir: ./saves/LLaMA2-7B/lora/train_24_8_13_08_13
|
||||||
logging_steps: 3
|
logging_steps: 3
|
||||||
save_steps: 100
|
save_steps: 100
|
||||||
plot_loss: true
|
plot_loss: true
|
||||||
|
|
Loading…
Reference in New Issue