config.json

{
  "model_type": "resnet",
  "num_classes": 15,
  "input_size": 224,
  "output_size": 256,
  "hidden_layers": 3,
  "dropout_prob": 0.4,
  "activation": "relu"
}