File size: 1,032 Bytes
f2f5dd2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
{
  "model_type": "emotion2vec_plus_large",
  "architecture": "data2vec2",
  "num_classes": 9,
  "hidden_dim": 1024,
  "ffn_dim": 4096,
  "num_layers": 12,
  "num_context_blocks": 4,
  "num_shared_blocks": 8,
  "num_heads": 16,
  "conv_feature_layers": [
    [
      1,
      512,
      10,
      5
    ],
    [
      512,
      512,
      3,
      2
    ],
    [
      512,
      512,
      3,
      2
    ],
    [
      512,
      512,
      3,
      2
    ],
    [
      512,
      512,
      3,
      2
    ],
    [
      512,
      512,
      2,
      2
    ],
    [
      512,
      512,
      2,
      2
    ]
  ],
  "pos_conv_depth": 5,
  "pos_conv_kernel": 19,
  "pos_conv_groups": 16,
  "feature_dim": 512,
  "has_context_norm": true,
  "has_extra_tokens": true,
  "has_alibi_scale": true,
  "num_extra_tokens": 10,
  "layer_norm_first": false,
  "dtype": "float16",
  "emotion_labels": [
    "angry",
    "disgusted",
    "fearful",
    "happy",
    "neutral",
    "other",
    "sad",
    "surprised",
    "unknown"
  ]
}