config.json

{
    "model_config": {
        "propagation_steps": 5,
        "encoder_config": {
            "node_hidden_sizes": [
                10
            ],
            "node_feature_dim": 1,
            "edge_hidden_sizes": null,
            "edge_feature_dim": 1
        },
        "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"
        },
        "aggregator_config": {
            "node_hidden_sizes": [
                10
            ],
            "graph_transform_sizes": [
                10
            ],
            "input_size": [
                10
            ],
            "gated": true,
            "aggregation_type": "sum"
        }
    },
    "name": "gmn_embed"
}