change: change llama2_lora_sft.yaml dataset to alpaca_zh

This commit is contained in:
wql 2024-08-12 15:47:08 +08:00
parent 90c6e4d020
commit 01f70612c7
1 changed files with 2 additions and 2 deletions

View File

@ -8,7 +8,7 @@ finetuning_type: lora
lora_target: all
### dataset
dataset: identity,alpaca_en_demo
dataset: alpaca_zh
template: llama2
cutoff_len: 1024
max_samples: 1000
@ -16,7 +16,7 @@ overwrite_cache: true
preprocessing_num_workers: 16
### output
output_dir: ./saves/LLaMA2-7B/lora/train_24_8_12_10_47
output_dir: ./saves/LLaMA2-7B/lora/train_24_8_12_15_46
logging_steps: 10
save_steps: 500
plot_loss: true