From ec40845513b1c8e6a2997c9eb1a57ca9182024f0 Mon Sep 17 00:00:00 2001 From: lwwang1995 Date: Tue, 8 Dec 2020 10:38:58 +0800 Subject: [PATCH] Update settings. --- examples/benchmarks/GATs/workflow_config_gats_Alpha158.yaml | 2 +- examples/benchmarks/GRU/workflow_config_gru_Alpha158.yaml | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/benchmarks/GATs/workflow_config_gats_Alpha158.yaml b/examples/benchmarks/GATs/workflow_config_gats_Alpha158.yaml index ff390ba07..71454e7f9 100644 --- a/examples/benchmarks/GATs/workflow_config_gats_Alpha158.yaml +++ b/examples/benchmarks/GATs/workflow_config_gats_Alpha158.yaml @@ -57,7 +57,7 @@ task: dropout: 0.7 n_epochs: 200 lr: 1e-4 - early_stop: 20 + early_stop: 10 metric: loss loss: mse base_model: LSTM diff --git a/examples/benchmarks/GRU/workflow_config_gru_Alpha158.yaml b/examples/benchmarks/GRU/workflow_config_gru_Alpha158.yaml index b52ed4160..13c2794bf 100755 --- a/examples/benchmarks/GRU/workflow_config_gru_Alpha158.yaml +++ b/examples/benchmarks/GRU/workflow_config_gru_Alpha158.yaml @@ -57,7 +57,7 @@ task: num_layers: 2 dropout: 0.0 n_epochs: 200 - lr: 1e-3 + lr: 2e-4 early_stop: 10 batch_size: 800 metric: loss