config.json

{
  "architectures": ["ResNetForImageClassification"],
  "model_type": "resnet",
  "framework": "pytorch",
  "num_labels": 4,
  "image_size": 224
}