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