Create config.json

config.json

@@ -0,0 +1,16 @@
+{
+  "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
+}