Sakibrumu commited on
Commit
1410d6a
1 Parent(s): 289bfab

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +9 -8
config.json CHANGED
@@ -1,8 +1,9 @@
1
- {
2
- "model_architecture": "ResNet50",
3
- "num_classes": 10,
4
- "input_size": [224, 224],
5
- "pretrained": true,
6
- "learning_rate": 0.001,
7
- "batch_size": 32
8
- }
 
 
1
+ {
2
+ "model_type": "resnet",
3
+ "model_name_or_path": "Sakibrumu/Food_Image_Classifier",
4
+ "framework": "pt",
5
+ "use_auth_token": false,
6
+ "input_size": 224,
7
+ "num_classes": 10,
8
+ "architecture": "resnet50"
9
+ }