chore: 调整训练 sh 脚本

This commit is contained in:
unanmed 2026-04-28 17:51:02 +08:00
parent 6460d8c5bc
commit 3e962bf8be

View File

@ -20,12 +20,12 @@ TRAIN_DATA="ginka-dataset.json"
EVAL_DATA="ginka-eval.json"
# 阶段 0预训练
P0_EPOCHS=50
P0_EPOCHS=100
P0_CHECKPOINT=10
P0_FINAL="result/pretrain/pretrain_final.pth"
# 阶段 1冻结编码器热身
P1_EPOCHS=30
P1_EPOCHS=50
P1_CHECKPOINT=10
P1_FINAL="result/joint/warmup_final.pth"