多任务微调时默认不加载预训练时quantified_delivery的预测头参数

2026-03-21 09:36:32 +08:00 · 2026-02-28 11:13:39 +08:00 · 2026-02-28 11:13:39 +08:00 · 60e082af55
commit 60e082af55
parent c52b82786d
2 changed files with 2 additions and 2 deletions
--- a/lnp_ml/modeling/final_train_optuna_cv.py
+++ b/lnp_ml/modeling/final_train_optuna_cv.py
@ -404,7 +404,7 @@ def main(
    swa_start_ratio: float = 0.75,
    # 预训练权重
    init_from_pretrain: Optional[Path] = None,
-    load_delivery_head: bool = True,
+    load_delivery_head: bool = False,
    # MPNN
    use_mpnn: bool = False,
    # 设备
--- a/lnp_ml/modeling/nested_cv_optuna.py
+++ b/lnp_ml/modeling/nested_cv_optuna.py
@ -522,7 +522,7 @@ def main(
    batch_size: int = 32,
    # 预训练权重
    init_from_pretrain: Optional[Path] = None,
-    load_delivery_head: bool = True,
+    load_delivery_head: bool = False,
    # MPNN
    use_mpnn: bool = False,
    # 设备