nickmuchi commited on
Commit
d23b801
1 Parent(s): 08996f4

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +8 -0
config.json CHANGED
@@ -14,12 +14,20 @@
14
  "hidden_act": "gelu",
15
  "hidden_dropout_prob": 0.0,
16
  "hidden_size": 384,
 
 
 
 
17
  "image_size": [
18
  512,
19
  864
20
  ],
21
  "initializer_range": 0.02,
22
  "intermediate_size": 1536,
 
 
 
 
23
  "layer_norm_eps": 1e-12,
24
  "model_type": "yolos",
25
  "num_attention_heads": 6,
 
14
  "hidden_act": "gelu",
15
  "hidden_dropout_prob": 0.0,
16
  "hidden_size": 384,
17
+ "id2label": {
18
+ "0": "name",
19
+ "1": "license-plates"
20
+ },
21
  "image_size": [
22
  512,
23
  864
24
  ],
25
  "initializer_range": 0.02,
26
  "intermediate_size": 1536,
27
+ "label2id": {
28
+ "name": 0,
29
+ "license-plates": 1
30
+ },
31
  "layer_norm_eps": 1e-12,
32
  "model_type": "yolos",
33
  "num_attention_heads": 6,