| { |
| "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" |
| } |
