config.json

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