config.json

{
    "model_type": "swin",
    "architectures": [
        "SwinForImageClassification"
    ],
    "num_classes": 14,
    "image_size": 512,
    "patch_size": 4,
    "embed_dim": 128,
    "depths": [
        2,
        2,
        18,
        2
    ],
    "num_heads": [
        4,
        8,
        16,
        32
    ],
    "window_size": 7
}