AncTe/ginka-generator
1
0
Fork 0
mirror of https://github.com/unanmed/ginka-generator.git synced 2026-05-22 19:31:12 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

chore: 调整训练 sh 脚本

Browse Source
This commit is contained in:
unanmed 2026-04-28 17:51:02 +08:00
parent 6460d8c5bc
commit 3e962bf8be
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
train_full.sh
View File

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