Fix bugs in the new models

2021-03-24 21:17:55 +08:00
parent 15dda79e3b
commit e7467fd474
4 changed files with 67 additions and 60 deletions
--- a/configs/qlib/workflow_config_transformer_basic_Alpha360.yaml
+++ b/configs/qlib/workflow_config_transformer_basic_Alpha360.yaml
@@ -47,13 +47,12 @@ task:
            net_config:
                name: basic
                d_feat: 6
-                stem_dim: 48
-                embed_dims: [48, 48, 48, 48, 48]
+                embed_dim: 48
                num_heads: [4, 4, 4, 4, 4]
                mlp_hidden_multipliers: [4, 4, 4, 4, 4]
                qkv_bias: True
                pos_drop: 0.1
-                other_drop: 0.1
+                other_drop: 0
            opt_config:
            loss: mse
            GPU: 0