From a7db8ffc151d018d8dd9266587b9dff0995a6f65 Mon Sep 17 00:00:00 2001 From: RYDE-WORK Date: Sat, 28 Feb 2026 17:32:49 +0800 Subject: [PATCH] =?UTF-8?q?=E7=BC=A9=E5=B0=8Fd=5Fmodel/num=5Fheads/n=5Fatt?= =?UTF-8?q?n=5Flayers/hidden=5Fdim=E6=88=96=E8=80=85=E5=A2=9E=E5=A4=A7?= =?UTF-8?q?=E6=AD=A3=E5=88=99=E5=8C=96=E5=90=8E,=E6=AC=A0=E6=8B=9F?= =?UTF-8?q?=E5=90=88,=E6=81=A2=E5=A4=8D=E9=BB=98=E8=AE=A4=E5=8F=82?= =?UTF-8?q?=E6=95=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- lnp_ml/modeling/benchmark.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/lnp_ml/modeling/benchmark.py b/lnp_ml/modeling/benchmark.py index 3ad7ac9..6ac6b20 100644 --- a/lnp_ml/modeling/benchmark.py +++ b/lnp_ml/modeling/benchmark.py @@ -284,11 +284,11 @@ def main( data_dir: Path = PROCESSED_DATA_DIR / "benchmark", output_dir: Path = MODELS_DIR / "benchmark", # 模型参数 - d_model: int = 128, - num_heads: int = 4, - n_attn_layers: int = 2, + d_model: int = 256, + num_heads: int = 8, + n_attn_layers: int = 4, fusion_strategy: str = "attention", - head_hidden_dim: int = 64, + head_hidden_dim: int = 128, dropout: float = 0.1, # MPNN 参数 use_mpnn: bool = False,