File size: 1,313 Bytes
b209e3c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 |
{
"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"
} |