LagrangianMGN / config.json
yushuang88's picture
Update model configuration
4c3eb8c verified
Raw
History Blame Contribute Delete
528 Bytes
{
"_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
}