Update single_dcu_train.sh

3dd62397 · wanglch · 8653b7cf · 3dd62397
Commit 3dd62397 authored May 09, 2024 by wanglch
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

single_dcu_train.sh single_dcu_train.sh +1 -1

No files found.
--- a/single_dcu_train.sh
+++ b/single_dcu_train.sh
@@ -8,7 +8,7 @@ CUDA_VISIBLE_DEVICES=0 python ../FinGPT/LLaMA-Factory-main/src/train_bash.py \
    --dataset_dir ../../data \
    --template default \
    --finetuning_type lora \
-    --lora_target q_proj,v_proj \
+    --lora_target all \
    --output_dir ../../saves/FinGPT/lora/sft \
    --overwrite_cache \
    --overwrite_output_dir \