dwililiya commited on
Commit
6c00f0b
1 Parent(s): 85ae2ea

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +24 -24
config.json CHANGED
@@ -1,24 +1,24 @@
1
- {
2
- "model_type": "custom_model",
3
- "num_labels": 6,
4
- "label2id": {
5
- "BacterialBlights": 0,
6
- "Healthy": 1,
7
- "Mosaic": 2,
8
- "RedRot": 3,
9
- "Rust": 4,
10
- "Yellow": 5
11
- },
12
- "id2label": {
13
- "0": "BacterialBlights",
14
- "1": "Healthy",
15
- "2": "Mosaic",
16
- "3": "RedRot",
17
- "4": "Rust",
18
- "5": "Yellow"
19
- },
20
- "input_size": [3, 256, 256],
21
- "output_size": 6,
22
- "architecture": "EfficientNetB1",
23
- "license": "Community Data License Agreement"
24
- }
 
1
+ {
2
+ "model_type": "efficientnet",
3
+ "num_labels": 6,
4
+ "label2id": {
5
+ "BacterialBlights": 0,
6
+ "Healthy": 1,
7
+ "Mosaic": 2,
8
+ "RedRot": 3,
9
+ "Rust": 4,
10
+ "Yellow": 5
11
+ },
12
+ "id2label": {
13
+ "0": "BacterialBlights",
14
+ "1": "Healthy",
15
+ "2": "Mosaic",
16
+ "3": "RedRot",
17
+ "4": "Rust",
18
+ "5": "Yellow"
19
+ },
20
+ "input_size": [3, 256, 256],
21
+ "output_size": 6,
22
+ "architecture": "EfficientNetB1",
23
+ "license": "Community Data License Agreement"
24
+ }