tzhao3 commited on
Commit
bd96c0b
1 Parent(s): 4622f21

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +2 -4
config.json CHANGED
@@ -8,13 +8,11 @@
8
  "hidden_act": "gelu",
9
  "hidden_dropout_prob": 0.0,
10
  "hidden_size": 1024,
11
- "id2label": {'0': 'bed', '1': 'chair', '2':'couch', '3':'table', '4': 'wardrobe', '5': 'clock', '6': 'keyboard','7': 'lamp',
12
- '8': 'telephone', '9': 'television'},
13
  "image_size": 224,
14
  "initializer_range": 0.02,
15
  "intermediate_size": 4096,
16
- "label2id": {'bed': 0, 'chair': 1, 'couch': 2, 'table': 3, 'wardrobe': 4, 'clock': 5, 'keyboard': 6, 'lamp': 7,
17
- 'telephone': 8, 'television': 9},
18
  "layer_norm_eps": 1e-12,
19
  "model_type": "vit",
20
  "num_attention_heads": 16,
 
8
  "hidden_act": "gelu",
9
  "hidden_dropout_prob": 0.0,
10
  "hidden_size": 1024,
11
+ "id2label": {"0": "bed", "1": "chair", "2":"couch", "3":"table", "4":"wardrobe", "5":"clock", "6":"keyboard", "7":"lamp", "8":"telephone", "9":"television'},
 
12
  "image_size": 224,
13
  "initializer_range": 0.02,
14
  "intermediate_size": 4096,
15
+ "label2id": {'bed":"0", "chair":"1", "couch":"2", "table":"3", "wardrobe":"4", "clock":"5", "keyboard":"6", "lamp":"7", "telephone":"8", "television":"9"},
 
16
  "layer_norm_eps": 1e-12,
17
  "model_type": "vit",
18
  "num_attention_heads": 16,