train: change yaml

This commit is contained in:
wql 2024-08-13 09:10:03 +08:00
parent 2827dd1c98
commit 0841a0832f
1 changed files with 2 additions and 2 deletions

View File

@ -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