Fix models.

2026-07-04 03:21:00 +08:00 · 2020-12-06 22:43:52 +08:00
parent 412c9eee2e
commit 8d42092a7e
9 changed files with 45 additions and 48 deletions
--- a/examples/benchmarks/GRU/csi300_gru_ts.pkl
+++ b/examples/benchmarks/GRU/csi300_gru_ts.pkl
--- a/examples/benchmarks/GRU/workflow_config_gru_Alpha158.yaml
+++ b/examples/benchmarks/GRU/workflow_config_gru_Alpha158.yaml
@@ -56,7 +56,7 @@ task:
            hidden_size: 64
            num_layers: 2
            dropout: 0.0
-            n_epochs: 2
+            n_epochs: 200
            lr: 1e-3
            early_stop: 10
            batch_size: 800
--- a/examples/benchmarks/LSTM/csi300_lstm_ts.pkl
+++ b/examples/benchmarks/LSTM/csi300_lstm_ts.pkl