RYDE-WORK/lnp_ml
1
0
Fork 0
mirror of https://github.com/RYDE-WORK/lnp_ml.git synced 2026-03-21 09:36:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
RYDE-WORK 2026-02-28 11:13:39 +08:00
parent c52b82786d
commit 60e082af55
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lnp_ml/modeling/final_train_optuna_cv.py
View File

@ -404,7 +404,7 @@ def main(
swa_start_ratio: float = 0.75, swa_start_ratio: float = 0.75,
# 预训练权重 # 预训练权重
init_from_pretrain: Optional[Path] = None, init_from_pretrain: Optional[Path] = None,
load_delivery_head: bool = True, load_delivery_head: bool = False,
# MPNN # MPNN
use_mpnn: bool = False, use_mpnn: bool = False,
# 设备 # 设备

2
lnp_ml/modeling/nested_cv_optuna.py
View File

@ -522,7 +522,7 @@ def main(
batch_size: int = 32, batch_size: int = 32,
# 预训练权重 # 预训练权重
init_from_pretrain: Optional[Path] = None, init_from_pretrain: Optional[Path] = None,
load_delivery_head: bool = True, load_delivery_head: bool = False,
# MPNN # MPNN
use_mpnn: bool = False, use_mpnn: bool = False,
# 设备 # 设备