HedronCreeper commited on
Commit
7f74b4c
·
verified ·
1 Parent(s): a4574e7

Upload MobileNetV2ForImageClassification

Browse files
Files changed (1) hide show
  1. config.json +9 -2
config.json CHANGED
@@ -10,8 +10,16 @@
10
  "finegrained_output": true,
11
  "first_layer_is_expansion": true,
12
  "hidden_act": "relu6",
 
 
 
 
13
  "image_size": 224,
14
  "initializer_range": 0.02,
 
 
 
 
15
  "layer_norm_eps": 0.001,
16
  "min_depth": 8,
17
  "model_type": "mobilenet_v2",
@@ -20,6 +28,5 @@
20
  "problem_type": "single_label_classification",
21
  "semantic_loss_ignore_index": 255,
22
  "tf_padding": true,
23
- "transformers_version": "5.0.0",
24
- "use_cache": false
25
  }
 
10
  "finegrained_output": true,
11
  "first_layer_is_expansion": true,
12
  "hidden_act": "relu6",
13
+ "id2label": {
14
+ "0": "apple",
15
+ "1": "banana"
16
+ },
17
  "image_size": 224,
18
  "initializer_range": 0.02,
19
+ "label2id": {
20
+ "apple": 0,
21
+ "banana": 1
22
+ },
23
  "layer_norm_eps": 0.001,
24
  "min_depth": 8,
25
  "model_type": "mobilenet_v2",
 
28
  "problem_type": "single_label_classification",
29
  "semantic_loss_ignore_index": 255,
30
  "tf_padding": true,
31
+ "transformers_version": "5.0.0"
 
32
  }