Jamal44 commited on
Commit
917f9d5
1 Parent(s): c628d10

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +3 -6
config.json CHANGED
@@ -1,8 +1,5 @@
1
  {
2
- "model_type": "alexnet",
3
- "num_classes": 10,
4
- "img_width": 224,
5
- "img_height": 224,
6
- "mean": [0.485, 0.456, 0.406],
7
- "std": [0.229, 0.224, 0.225]
8
  }
 
1
  {
2
+ "architectures": ["AlexNet"],
3
+ "class_name": "AlexNet",
4
+ "num_classes": 10
 
 
 
5
  }