Upload config.json

config.json

@@ -1,13 +1,11 @@
 {
-    "model_type": "resnet",
+    "model_type": "resnet152_binary_classification",
     "input_shape": [224, 224, 3],
     "num_classes": 2,
-    "optimizer": {
-        "name": "Adam",
-        "learning_rate": 0.001
-    },
-    "batch_size": 32,
-    "epochs": 10,
-    "loss_function": "binary_crossentropy",
-    "metrics": ["accuracy"]
+    "layers": [
+        {"type": "resnet152", "output_shape": [7, 7, 2048]},
+        {"type": "global_average_pooling2d", "output_shape": [2048]},
+        {"type": "dense", "output_shape": [1]}
+    ],
+    "total_parameters": 58377093
 }