config.json

{
  "model_type": "custom_cnn",
  "num_classes": 15,
  "img_size": 224,
  "framework": "pytorch"
}