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
|
||||
|
||||
### dataset
|
||||
dataset: alpaca_gpt4_zh
|
||||
dataset: alpaca_en
|
||||
template: llama2
|
||||
cutoff_len: 1024
|
||||
max_samples: 100000
|
||||
|
@ -16,7 +16,7 @@ overwrite_cache: true
|
|||
preprocessing_num_workers: 16
|
||||
|
||||
### 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
|
||||
save_steps: 100
|
||||
plot_loss: true
|
||||
|
|
Loading…
Reference in New Issue