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