config.json

{
    "model_type": "resnet50",
    "pretrained": true,
    "num_classes": 6,
    "input_shape": [
        384,
        384,
        3
    ],
    "learning_rate": 0.0001,
    "optimizer": "adam",
    "loss_function": "cross_entropy"
}