config.json

{
  "ape": false,
  "attn_drop_rate": 0.0,
  "depths": [
    2,
    2,
    18,
    2
  ],
  "drop_path_rate": 0.2,
  "drop_rate": 0.0,
  "embed_dim": 128,
  "img_size": 256,
  "in_chans": 3,
  "mlp_ratio": 4.0,
  "num_classes": 1000,
  "num_heads": [
    4,
    8,
    16,
    32
  ],
  "output_dim": 512,
  "p": 3.0,
  "patch_norm": true,
  "patch_size": 4,
  "pretrained": null,
  "pretrained_window_sizes": [
    12,
    12,
    12,
    6
  ],
  "qkv_bias": true,
  "use_checkpoint": false,
  "window_size": 16
}