model/config.json

{
    "model_type": "resnet50",
    "num_classes": 9,
    "input_size": [
        3,
        28,
        28
    ],
    "architecture": "ResNet50"
}