{ "model_config": { "propagation_steps": 5, "encoder_config": { "node_hidden_sizes": [ 10 ], "node_feature_dim": 1, "edge_hidden_sizes": null, "edge_feature_dim": 1 }, "propagation_layer_config": { "node_state_dim": 10, "edge_hidden_sizes": [ 20 ], "node_hidden_sizes": [ 10 ], "edge_net_init_scale": 0.1, "node_update_type": "gru", "use_reverse_direction": true, "reverse_dir_param_different": false, "layer_norm": false, "prop_type": "embedding" }, "aggregator_config": { "node_hidden_sizes": [ 10 ], "graph_transform_sizes": [ 10 ], "input_size": [ 10 ], "gated": true, "aggregation_type": "sum" } }, "name": "gmn_embed" }