From 0841a0832f24e373867aaca572a30100fced7c9f Mon Sep 17 00:00:00 2001
From: wql <wengqinlan18@163.com>
Date: Tue, 13 Aug 2024 09:10:03 +0800
Subject: [PATCH] train: change yaml

---
 examples/train_lora/llama2_lora_sft.yaml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/examples/train_lora/llama2_lora_sft.yaml b/examples/train_lora/llama2_lora_sft.yaml
index fd7078e0..cb834840 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: 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