| { | |
| "model_config": { | |
| "propagation_layer_config": { | |
| "prop_type": "matching", | |
| "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 | |
| }, | |
| "propagation_steps": 5, | |
| "alignment_feature_dim": 16, | |
| "scoring": "aggregated", | |
| "interaction_alignment": "attention", | |
| "interaction_alignment_preprocessor_type": "identity", | |
| "interaction_when": "post", | |
| "encoder_config": { | |
| "node_hidden_sizes": [ | |
| 10 | |
| ], | |
| "node_feature_dim": 1, | |
| "edge_hidden_sizes": null, | |
| "edge_feature_dim": 1 | |
| }, | |
| "attention_config": { | |
| "temperature": 0.1 | |
| }, | |
| "aggregator_config": { | |
| "node_hidden_sizes": [ | |
| 10 | |
| ], | |
| "graph_transform_sizes": [ | |
| 10 | |
| ], | |
| "input_size": [ | |
| 10 | |
| ], | |
| "gated": true, | |
| "aggregation_type": "sum" | |
| } | |
| }, | |
| "name": "gmn_baseline_scoring=agg___tp=attention_pp=identity_when=post" | |
| } |