config.json

{
  "attn_drop_rate": 0.0,
  "ca_attentions": [
    1,
    1,
    1,
    0
  ],
  "ca_num_heads": [
    4,
    4,
    4,
    -1
  ],
  "depths": [
    2,
    2,
    8,
    1
  ],
  "drop_path_rate": 0.2,
  "drop_rate": 0.0,
  "embed_dims": [
    64,
    128,
    256,
    512
  ],
  "expand_ratio": 2,
  "head_conv": 3,
  "img_size": 512,
  "in_chans": 3,
  "layerscale_value": 0.0001,
  "mlp_ratios": [
    4,
    4,
    4,
    2
  ],
  "num_classes": 2,
  "num_stages": 4,
  "qk_scale": null,
  "qkv_bias": true,
  "sa_num_heads": [
    -1,
    -1,
    8,
    16
  ],
  "use_layerscale": false
}