Fix small bugs

This commit is contained in:
D-X-Y
2021-08-14 16:01:07 -07:00
parent 58733c18be
commit d04edcd211
12 changed files with 95 additions and 18 deletions

View File

@@ -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