Skip to content

GitLab

Commit f3aee530 authored by wanglch's avatar wanglch
Browse files

Update finetune_ds.sh

parent b7fdc1ae
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
finetune/finetune_ds.sh
View file @ f3aee530
...@@ -46,7 +46,7 @@ torchrun $DISTRIBUTED_ARGS finetune.py \ ...@@ -46,7 +46,7 @@ torchrun $DISTRIBUTED_ARGS finetune.py \
--gradient_accumulation_steps 1 \ --gradient_accumulation_steps 1 \
--evaluation_strategy "steps" \ --evaluation_strategy "steps" \
--save_strategy "steps" \ --save_strategy "steps" \
--save_steps 1000 \ --save_steps 100 \
--save_total_limit 10 \ --save_total_limit 10 \
--learning_rate 1e-6 \ --learning_rate 1e-6 \
--weight_decay 0.1 \ --weight_decay 0.1 \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment