| { | |
| "model_config": { | |
| "propagation_layer_config": { | |
| "prop_type": "embedding", | |
| "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, | |
| "refinement_steps": 3, | |
| "alignment_feature_dim": 16, | |
| "unify_scoring_and_interaction_preprocessor": true, | |
| "scoring": "set_aligned", | |
| "scoring_alignment": "sinkhorn", | |
| "scoring_alignment_preprocessor_type": "lrl", | |
| "interaction_alignment": "sinkhorn", | |
| "interaction_alignment_preprocessor_type": "lrl", | |
| "interaction_when": "pre", | |
| "encoder_config": { | |
| "node_hidden_sizes": [ | |
| 10 | |
| ], | |
| "node_feature_dim": 1, | |
| "edge_hidden_sizes": null, | |
| "edge_feature_dim": 1 | |
| }, | |
| "sinkhorn_config": { | |
| "noise_factor": 0 | |
| } | |
| }, | |
| "name": "gmn_iterative_refinement_scoring=sinkhorn_pp=lrl___tp=sinkhorn_pp=lrl_when=pre___unify=true" | |
| } |