Skip to content

Commit 064b2f2

Browse files
authored
Update learning rate.
1 parent 9edef40 commit 064b2f2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

scripts/train.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ torchrun --nproc_per_node=4 train.py \
2121
--eval_strategy "no" \
2222
--save_strategy "steps" \
2323
--save_steps 2000 \
24-
--learning_rate 1e-4 \
24+
--learning_rate 5e-6 \
2525
--weight_decay 0. \
2626
--warmup_ratio 0.03 \
2727
--lr_scheduler_type "cosine" \

0 commit comments

Comments
 (0)