mirror of
https://github.com/unanmed/ginka-generator.git
synced 2026-05-14 04:41:12 +08:00
chore: 调整训练 sh 脚本
This commit is contained in:
parent
6460d8c5bc
commit
3e962bf8be
@ -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"
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user