Andrazp commited on
Commit
7326d45
1 Parent(s): 23c2a58

updating config.json

Browse files
Files changed (1) hide show
  1. config.json +15 -14
config.json CHANGED
@@ -2,37 +2,38 @@
2
  "architectures": [
3
  "BertForSequenceClassification"
4
  ],
5
- "label2id": {
6
- "acceptable": 0,
7
- "innapropriate": 1,
8
- "offensive": 2,
9
- "violent": 3
10
- },
11
  "attention_probs_dropout_prob": 0.1,
12
  "directionality": "bidi",
13
- "finetuning_task": null,
14
  "hidden_act": "gelu",
15
  "hidden_dropout_prob": 0.1,
16
  "hidden_size": 768,
 
 
 
 
 
 
17
  "initializer_range": 0.02,
18
  "intermediate_size": 3072,
19
- "is_decoder": false,
 
 
 
 
 
20
  "layer_norm_eps": 1e-12,
21
  "max_position_embeddings": 512,
 
22
  "num_attention_heads": 12,
23
  "num_hidden_layers": 12,
24
- "num_labels": 4,
25
- "output_attentions": false,
26
- "output_hidden_states": false,
27
  "output_past": true,
 
28
  "pooler_fc_size": 768,
29
  "pooler_num_attention_heads": 12,
30
  "pooler_num_fc_layers": 3,
31
  "pooler_size_per_head": 128,
32
  "pooler_type": "first_token_transform",
33
- "pruned_heads": {},
34
- "torchscript": false,
35
  "type_vocab_size": 2,
36
- "use_bfloat16": false,
37
  "vocab_size": 49601
38
  }
2
  "architectures": [
3
  "BertForSequenceClassification"
4
  ],
 
 
 
 
 
 
5
  "attention_probs_dropout_prob": 0.1,
6
  "directionality": "bidi",
7
+ "gradient_checkpointing": false,
8
  "hidden_act": "gelu",
9
  "hidden_dropout_prob": 0.1,
10
  "hidden_size": 768,
11
+ "id2label": {
12
+ "0": "LABEL_0",
13
+ "1": "LABEL_1",
14
+ "2": "LABEL_2",
15
+ "3": "LABEL_3"
16
+ },
17
  "initializer_range": 0.02,
18
  "intermediate_size": 3072,
19
+ "label2id": {
20
+ "LABEL_0": 0,
21
+ "LABEL_1": 1,
22
+ "LABEL_2": 2,
23
+ "LABEL_3": 3
24
+ },
25
  "layer_norm_eps": 1e-12,
26
  "max_position_embeddings": 512,
27
+ "model_type": "bert",
28
  "num_attention_heads": 12,
29
  "num_hidden_layers": 12,
 
 
 
30
  "output_past": true,
31
+ "pad_token_id": 0,
32
  "pooler_fc_size": 768,
33
  "pooler_num_attention_heads": 12,
34
  "pooler_num_fc_layers": 3,
35
  "pooler_size_per_head": 128,
36
  "pooler_type": "first_token_transform",
 
 
37
  "type_vocab_size": 2,
 
38
  "vocab_size": 49601
39
  }