From f7e375ae6803f8f408fd14721530a2daabd99220 Mon Sep 17 00:00:00 2001 From: Haoyu Wang <48108414+javaThonc@users.noreply.github.com> Date: Thu, 26 Nov 2020 16:08:21 +0800 Subject: [PATCH] Update workflow_config_sfm.yaml --- examples/benchmarks/SFM/workflow_config_sfm.yaml | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/examples/benchmarks/SFM/workflow_config_sfm.yaml b/examples/benchmarks/SFM/workflow_config_sfm.yaml index 743c6220f..04f796150 100644 --- a/examples/benchmarks/SFM/workflow_config_sfm.yaml +++ b/examples/benchmarks/SFM/workflow_config_sfm.yaml @@ -31,21 +31,21 @@ task: kwargs: d_feat: 6 hidden_size: 64 - output_dim: 1 - freq_dim: 20 + output_dim: 32 + freq_dim: 25 dropout_W: 0.5 dropout_U: 0.5 - n_epochs: 10 + n_epochs: 20 lr: 1e-3 - batch_size: 800 + batch_size: 1600 early_stop: 20 eval_steps: 5 loss: mse lr_decay: 0.96 lr_decay_steps: 100 - optimizer: gd + optimizer: adam GPU: 1 - seed: 0 + seed: 710 dataset: class: DatasetH module_path: qlib.data.dataset