config.json

{
    "model": "efficientnet_b0",
    "num_labels": 3,
    "classes": [
        "brain_glioma",
        "brain_menin",
        "brain_tumor"
    ],
    "image_size": [
        224,
        224
    ]
}