config.json

{
    "model_checkpoint" : "SJChaudhuri/Efficient-MaxViT",
    "num_layers": 12,
    "hidden_size": 768,
    "num_heads": 12,
    "mlp_dim": 3072,
    "num_classes": 5,
    "input_image_size": [
        224,
        224
    ],
    "patch_size": 16,
    "vocab_size": 30000,
    "embedding_dropout": 0.1,
    "attention_dropout": 0.1,
    "dropout": 0.1,
    "num_epochs": 50,
    "batch_size": 32,
    "learning_rate": 0.001,
    "optimizer": "AdamW",
    "class_mapping": {
        "RG-0": "a_noDR",
        "RG-1": "b_mildDR",
        "RG-2": "c_moderateDR",
        "RG-3": "d_severeDR",
        "RG-4": "e_proDR"
    }
}