u2003158 commited on
Commit
296ca35
1 Parent(s): 65f9752

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -5,5 +5,5 @@
5
  "model_type": "resnet",
6
  "model_name": "resnet50",
7
  "weights": "imagenet",
8
- "num_classes": 5
9
  }
 
5
  "model_type": "resnet",
6
  "model_name": "resnet50",
7
  "weights": "imagenet",
8
+ "num_classes": 1000
9
  }