config.json

{
  "architectures": [
    "RSPResNetForImageClassification"
  ],
  "auto_map": {
    "AutoConfig": "configuration_rsp.RSPResNetConfig",
    "AutoModelForImageClassification": "modeling_rsp.RSPResNetForImageClassification"
  },
  "block": "Bottleneck",
  "image_size": 224,
  "layers": [
    3,
    4,
    6,
    3
  ],
  "model_type": "rsp_resnet",
  "num_channels": 3,
  "num_labels": 51,
  "source_checkpoint": "/mnt/data/projects/model_hubs/raw/rsp-resnet-50-ckpt.pth"
}