AJosh commited on
Commit
ca0ba3c
·
verified ·
1 Parent(s): 2a15953

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +4 -9
config.json CHANGED
@@ -1,13 +1,8 @@
1
  {
2
  "model_type": "vit",
3
  "num_labels": 7,
4
- "id2label": {
5
- "0": "Angry",
6
- "1": "Disgust",
7
- "2": "Fear",
8
- "3": "Happy",
9
- "4": "Sad",
10
- "5": "Surprise",
11
- "6": "Neutral"
12
- }
13
  }
 
1
  {
2
  "model_type": "vit",
3
  "num_labels": 7,
4
+ "hidden_size": 768,
5
+ "num_hidden_layers": 12,
6
+ "num_attention_heads": 12,
7
+ "intermediate_size": 3072
 
 
 
 
 
8
  }