203213d b6f7d03 203213d
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
{ "in_dim": 10, "out_dim": 1, "hidden_dim": 64, "num_layers": 4, "num_heads": 4, "space_dim": 2, "include_pos": true, "dropout": 0.0, "activation": "gelu", "mlp_ratio": 1, "attn_type": "nystrom", "psi_dim": 8 }
