config.json

{
  "model_type": "custom_pytorch_model",
  "num_classes": 8,
  "height": 224,
  "width": 224,
  "num_input_channels": 3,
  "num_cnn_channels": 16,
  "num_cnn_layers": 3,
  "hidden_dim": 16,
  "num_layers": 1,
  "kernel_size": 3,
  "stride": 1,
  "padding": 1,
  "pooling": 2,
  "dropout": 0.2
}