From bbc7c88f99e0c2b7095d15a4a755adb989414019 Mon Sep 17 00:00:00 2001 From: RYDE-WORK Date: Sat, 28 Feb 2026 16:17:40 +0800 Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=A4=A7benchmark=E6=97=B6=E7=9A=84?= =?UTF-8?q?=E5=9B=BA=E5=AE=9Adropout=E5=92=8Cweight=20decay?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- lnp_ml/modeling/benchmark.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lnp_ml/modeling/benchmark.py b/lnp_ml/modeling/benchmark.py index d023170..ec8d9fe 100644 --- a/lnp_ml/modeling/benchmark.py +++ b/lnp_ml/modeling/benchmark.py @@ -289,7 +289,7 @@ def main( n_attn_layers: int = 4, fusion_strategy: str = "attention", head_hidden_dim: int = 128, - dropout: float = 0.1, + dropout: float = 0.3, # MPNN 参数 use_mpnn: bool = False, mpnn_checkpoint: Optional[str] = None, @@ -298,7 +298,7 @@ def main( # 训练参数 batch_size: int = 64, lr: float = 1e-4, - weight_decay: float = 1e-5, + weight_decay: float = 1e-3, epochs: int = 50, patience: int = 10, # 设备