peterhuang24 commited on
Commit
5e0a5c4
1 Parent(s): 6779f09

Upload config.json

Browse files
Files changed (1) hide show
  1. config.json +6 -9
config.json CHANGED
@@ -1,11 +1,8 @@
1
  {
2
- "model_type": "resnet",
3
- "input_shape": [224, 224, 3],
4
- "num_classes": 2,
5
- "layers": [
6
- {"type": "resnet152", "output_shape": [7, 7, 2048]},
7
- {"type": "global_average_pooling2d", "output_shape": [2048]},
8
- {"type": "dense", "output_shape": [1]}
9
- ],
10
- "total_parameters": 58377093
11
  }
 
1
  {
2
+ "model_type": "resnet",
3
+ "input_shape": [224, 224, 3],
4
+ "num_classes": 1,
5
+ "architecture": "ResNet152",
6
+ "description": "Fine-tuned ResNet152 for binary classification"
7
+
 
 
 
8
  }