bug fix

Author: Taka008

pretrain/scripts/v3-1.7b-exp2-cpt-2epoch-sakura/train.sh

@@ -54,7 +54,7 @@ GRAD_CLIP=1
 # 988240 + 988240 = 1976480
 LR_WARMUP_STEPS=2000
 LR_DECAY_ITERS=988240
-TRAIN_STEPS=$(((${LR_WARMUP_STEPS} + ${LR_DECAY_ITERS}) * 2)
+TRAIN_STEPS=$(((${LR_WARMUP_STEPS} + ${LR_DECAY_ITERS}) * 2))
 
 # model config
 TOKENIZER_MODEL=${ENV_DIR}/src/llm-jp-tokenizer/models/ver3.0/llm-jp-tokenizer-100k.ver3.0b1.model