diff --git a/lnp_ml/modeling/final_train_optuna_cv.py b/lnp_ml/modeling/final_train_optuna_cv.py index a439b82..4eeb0e7 100644 --- 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, # 设备 diff --git a/lnp_ml/modeling/nested_cv_optuna.py b/lnp_ml/modeling/nested_cv_optuna.py index dc78aef..0d5ba11 100644 --- 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, # 设备