| { | |
| "name": "MeshGraphNet", | |
| "specific_params": { | |
| "MeshGraphNet": { | |
| "num_input_features": 6, | |
| "num_edge_features": 3, | |
| "num_output_features": 3, | |
| "processor_size": 2, | |
| "hidden_dim_processor": 32, | |
| "num_layers_node_processor": 2, | |
| "num_layers_edge_processor": 2, | |
| "hidden_dim_node_encoder": 32, | |
| "hidden_dim_edge_encoder": 32, | |
| "hidden_dim_node_decoder": 32, | |
| "do_concat_trick": false, | |
| "num_processor_checkpoint_segments": 0, | |
| "recompute_activation": false | |
| } | |
| } | |
| } | |