config.json

{
    "architectures": [
        "sequential"
    ],
    "task_type": "your_task_type",
    "input_info": [
        {
            "name": "conv2d",
            "shape": [
                null,
                224,
                224,
                3
            ]
        },
        {
            "name": "max_pooling2d",
            "shape": [
                null,
                222,
                222,
                32
            ]
        },
        {
            "name": "flatten",
            "shape": [
                null,
                111,
                111,
                32
            ]
        },
        {
            "name": "dense",
            "shape": [
                null,
                394272
            ]
        },
        {
            "name": "dense_1",
            "shape": [
                null,
                64
            ]
        }
    ],
    "output_info": [
        {
            "name": "dense_1/Softmax:0",
            "shape": [
                null,
                5
            ]
        }
    ]
}