| { | |
| "model_config": { | |
| "propagation_steps": 5, | |
| "sinkhorn_feature_dim": 16, | |
| "time_update_steps": 3, | |
| "propagation_layer_config": { | |
| "edge_embedding_dim": 20, | |
| "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" | |
| }, | |
| "encoder_config": { | |
| "edge_hidden_sizes": [ | |
| 20 | |
| ], | |
| "node_hidden_sizes": [ | |
| 10 | |
| ], | |
| "node_feature_dim": 1, | |
| "edge_feature_dim": 1 | |
| }, | |
| "interaction_when": "pre", | |
| "sinkhorn_config": { | |
| "noise_factor": 0 | |
| } | |
| }, | |
| "name": "edge_early_interaction" | |
| } |