Update finetune_qlora_ds.sh

This commit is contained in:
Junyang Lin
2023-10-08 14:50:45 +08:00
committed by GitHub
parent 14e6e69f6d
commit 26922b81cb

View File

@@ -35,7 +35,7 @@ torchrun $DISTRIBUTED_ARGS finetune.py \
--save_strategy "steps" \ --save_strategy "steps" \
--save_steps 1000 \ --save_steps 1000 \
--save_total_limit 10 \ --save_total_limit 10 \
--learning_rate 1e-5 \ --learning_rate 3e-4 \
--weight_decay 0.1 \ --weight_decay 0.1 \
--adam_beta2 0.95 \ --adam_beta2 0.95 \
--warmup_ratio 0.01 \ --warmup_ratio 0.01 \