diff --git a/train_full.sh b/train_full.sh index 1e27d7f..6c6135e 100644 --- a/train_full.sh +++ b/train_full.sh @@ -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"