File size: 639 Bytes
d8aee92 e9a5c29 d8aee92 e9a5c29 d8aee92 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 | {
"name": "Transolver",
"specific_params": {
"Transolver": {
"n_hidden": 32,
"n_layers": 2,
"space_dim": 7,
"fun_dim": 0,
"n_head": 4,
"mlp_ratio": 2,
"out_dim": 4,
"slice_num": 8,
"unified_pos": 0,
"build_graph": false,
"r": 0.05,
"max_neighbors": 32
},
"Transolver_plus": {
"n_hidden": 32,
"n_layers": 2,
"space_dim": 7,
"fun_dim": 0,
"n_head": 4,
"mlp_ratio": 2,
"out_dim": 4,
"slice_num": 8,
"unified_pos": 0,
"build_graph": false,
"r": 0.05,
"max_neighbors": 32
}
}
}
|