peterhuang24 commited on
Commit
713df86
·
verified ·
1 Parent(s): 937d8fb

upload config file.

Browse files
Files changed (1) hide show
  1. config.json +14 -0
config.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architecture": "ResNet152",
3
+ "description": "Custom classifier on top of ResNet152 for image classification",
4
+ "input_shape": [224, 224, 3],
5
+ "num_classes": 2,
6
+ "optimizer": {
7
+ "name": "Adam",
8
+ "learning_rate": 0.001
9
+ },
10
+ "batch_size": 32,
11
+ "epochs": 10,
12
+ "loss_function": "binary_crossentropy",
13
+ "metrics": ["accuracy"]
14
+ }