config.json

{
  "architectures": [
    "resnet50"
  ],
  "model_type": "resnet50",
  "num_classes": 10,
  "num_layers": 18,
  "hidden_size": 64,
  "dropout_prob": 0.5
}