{
    "model_type": "resnet",
    "architectures": ["ResNet"],
    "num_labels": 4,
    "image_size": 224,
    "input_size": [3, 224, 224]
  }