| { | |
| "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 | |
| } | |
| } | |
| } | |