We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 9edef40 commit 064b2f2Copy full SHA for 064b2f2
scripts/train.sh
@@ -21,7 +21,7 @@ torchrun --nproc_per_node=4 train.py \
21
--eval_strategy "no" \
22
--save_strategy "steps" \
23
--save_steps 2000 \
24
- --learning_rate 1e-4 \
+ --learning_rate 5e-6 \
25
--weight_decay 0. \
26
--warmup_ratio 0.03 \
27
--lr_scheduler_type "cosine" \
0 commit comments