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, # 设备