diff --git a/examples/train_lora/llama2_lora_sft.yaml b/examples/train_lora/llama2_lora_sft.yaml index 5f4c58cb..12f9d4d0 100644 --- a/examples/train_lora/llama2_lora_sft.yaml +++ b/examples/train_lora/llama2_lora_sft.yaml @@ -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