Remove batchsize and add daily-batch mode

2026-07-09 22:10:56 +08:00 · 2020-11-26 20:22:51 +08:00
parent 99efaadd38
commit ab98f44345
3 changed files with 30 additions and 31 deletions
--- a/examples/workflow_by_code_hats.py
+++ b/examples/workflow_by_code_hats.py
@@ -62,12 +62,11 @@ if __name__ == "__main__":
                "n_epochs": 200,
                "lr": 1e-3,
                "early_stop": 20,
-                "batch_size": 800,
                "metric": "IC",
                "loss": "mse",
                "base_model": "LSTM",
                "seed": 0,
-                "GPU": "0",
+                "GPU": "2",
            },
        },
        "dataset": {