| { | |
| "model_config": { | |
| "propagation_steps": 5, | |
| "sinkhorn_feature_dim": 16, | |
| "encoder_config": { | |
| "node_hidden_sizes": [ | |
| 10 | |
| ], | |
| "node_feature_dim": 1, | |
| "edge_hidden_sizes": null, | |
| "edge_feature_dim": 1 | |
| }, | |
| "sinkhorn_config": { | |
| "noise_factor": 0 | |
| }, | |
| "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" | |
| } | |
| }, | |
| "name": "node_align_node_loss" | |
| } |