Fix small bugs

2021-08-14 16:01:07 -07:00
parent 58733c18be
commit d04edcd211
12 changed files with 95 additions and 18 deletions
--- a/configs/archs/NAS-CIFAR-DARTS.config
+++ b/configs/archs/NAS-CIFAR-DARTS.config
@@ -0,0 +1,10 @@
+{
+  "arch"      : ["str",  "dxys"],
+  "genotype"  : ["str",  "DARTS"],
+  "dataset"   : ["str",  "cifar"],
+  "ichannel"  : ["int",   36],
+  "layers"    : ["int",    6],
+  "stem_multi": ["int",    3],
+  "auxiliary" : ["bool",   1],
+  "drop_path_prob": ["float", 0.2]
+}
--- a/configs/qlib/workflow_config_transformer_basic_Alpha360.yaml
+++ b/configs/qlib/workflow_config_transformer_basic_Alpha360.yaml
@@ -47,7 +47,7 @@ task:
            net_config:
                name: basic
                d_feat: 6
-                embed_dim: 48
+                embed_dim: 32
                num_heads: [4, 4, 4, 4, 4]
                mlp_hidden_multipliers: [4, 4, 4, 4, 4]
                qkv_bias: True