siddheshtv commited on
Commit
919e3a0
1 Parent(s): 4dd8f2f

update config

Browse files
Files changed (1) hide show
  1. config.json +2 -1
config.json CHANGED
@@ -1,5 +1,6 @@
1
  {
2
  "architectures": ["BlockNet10"],
3
  "hidden_size": 1024,
4
- "num_classes": 10
 
5
  }
 
1
  {
2
  "architectures": ["BlockNet10"],
3
  "hidden_size": 1024,
4
+ "num_classes": 10,
5
+ "model_type": "custom_model"
6
  }