File size: 528 Bytes
7834f88 4c3eb8c 7834f88 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | {
"_target_": "model.meshgraphnet.MeshGraphNet",
"_convert_": "all",
"input_dim_nodes": null,
"input_dim_edges": null,
"output_dim": null,
"processor_size": 2,
"aggregation": "sum",
"hidden_dim_processor": 32,
"hidden_dim_node_encoder": 64,
"hidden_dim_edge_encoder": 64,
"hidden_dim_node_decoder": 64,
"num_layers_node_processor": 2,
"num_layers_edge_processor": 2,
"mlp_activation_fn": "relu",
"do_concat_trick": false,
"num_processor_checkpoint_segments": 0,
"recompute_activation": false
}
|