config.json

{
  "model_type": "swin",
  "num_classes": 5,
  "num_features": 1024,
  "global_pool": "avg",
  "mean": [
    0.485,
    0.456,
    0.406
  ],
  "std": [
    0.229,
    0.224,
    0.225
  ],
  "input_size": [
    3,
    224,
    224
  ],
  "interpolation": "bicubic",
  "crop_pct": 0.9,
  "crop_mode": "center",
  "first_conv": "patch_embed.proj",
  "classifier": "head",
  "id2label": {
    "0": "content",
    "1": "end",
    "2": "start",
    "3": "subt",
    "4": "subtl"
  },
  "label2id": {
    "content": 0,
    "end": 1,
    "start": 2,
    "subt": 3,
    "subtl": 4
  }
}