config.json

{
    "architecture": "vit_base_patch16_224.augreg_in21k_ft_in1k",
    "input_size": [
        3,
        224,
        224
    ],
    "num_classes": 215
}