config.json

{
  "NC_depth": [
    2,
    2,
    8,
    2
  ],
  "NC_group": [
    1,
    32,
    64,
    128
  ],
  "NC_heads": [
    1,
    2,
    4,
    8
  ],
  "RC_group": [
    1,
    16,
    32,
    64
  ],
  "RC_heads": [
    1,
    1,
    2,
    4
  ],
  "architectures": [
    "RSPViTAEForImageClassification"
  ],
  "auto_map": {
    "AutoConfig": "configuration_rsp.RSPViTAEConfig",
    "AutoModelForImageClassification": "modeling_rsp.RSPViTAEForImageClassification"
  },
  "downsample_ratios": [
    4,
    2,
    2,
    2
  ],
  "embed_dims": [
    64,
    64,
    128,
    256
  ],
  "image_size": 224,
  "mlp_ratio": 4.0,
  "model_type": "rsp_vitae",
  "num_channels": 3,
  "num_labels": 51,
  "source_checkpoint": "/mnt/data/projects/model_hubs/raw/rsp-vitaev2-s-ckpt.pth",
  "stages": 4,
  "token_dims": [
    64,
    128,
    256,
    512
  ]
}