add arg weight_decay

2026-07-02 10:31:00 +08:00 · 2020-12-09 00:27:54 +08:00
parent 2873813562
commit 56e579e20f
2 changed files with 8 additions and 3 deletions
--- a/examples/benchmarks/MLP/workflow_config_mlp_Alpha158.yaml
+++ b/examples/benchmarks/MLP/workflow_config_mlp_Alpha158.yaml
@@ -67,6 +67,7 @@ task:
            max_steps: 8000
            batch_size: 8192
            GPU: 0
+            weight_decay: 0.0002
    dataset:
        class: DatasetH
        module_path: qlib.data.dataset