Add model_path param to gats

2026-07-10 22:36:55 +08:00 · 2020-11-30 22:26:59 +08:00
parent 620e325da1
commit d488a503b2
2 changed files with 23 additions and 8 deletions
--- a/examples/benchmarks/GATs/workflow_config_gats.yaml
+++ b/examples/benchmarks/GATs/workflow_config_gats.yaml
@@ -54,6 +54,8 @@ task:
            metric: loss
            loss: mse
            base_model: LSTM
+            with_pretrain: True
+            model_path: "benchmarks/LSTM/model_lstm_csi300.pkl"
            seed: 0
            GPU: 0
    dataset: