peterhuang24 commited on
Commit
46e609b
·
verified ·
1 Parent(s): 452ab01

Upload config.json

Browse files
Files changed (1) hide show
  1. config.json +1 -2
config.json CHANGED
@@ -1,6 +1,5 @@
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": {
 
1
  {
2
+ "model_type": "resnet",
 
3
  "input_shape": [224, 224, 3],
4
  "num_classes": 2,
5
  "optimizer": {