system commited on
Commit
9e7b6fe
1 Parent(s): d9fd1c8

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +12 -17
config.json CHANGED
@@ -1,36 +1,31 @@
1
  {
 
2
  "architectures": [
3
  "DistilBertForSequenceClassification"
4
  ],
5
- "activation": "gelu",
6
  "attention_dropout": 0.1,
7
  "dim": 768,
8
  "dropout": 0.1,
9
  "finetuning_task": "sst-2",
10
  "hidden_dim": 3072,
 
 
 
 
11
  "initializer_range": 0.02,
12
- "is_decoder": false,
 
 
 
13
  "max_position_embeddings": 512,
 
14
  "n_heads": 12,
15
  "n_layers": 6,
16
- "num_labels": 2,
17
- "output_attentions": false,
18
- "output_hidden_states": false,
19
  "output_past": true,
20
- "pruned_heads": {},
21
  "qa_dropout": 0.1,
22
  "seq_classif_dropout": 0.2,
23
  "sinusoidal_pos_embds": false,
24
  "tie_weights_": true,
25
- "torchscript": false,
26
- "use_bfloat16": false,
27
- "vocab_size": 30522,
28
- "label2id": {
29
- "NEGATIVE": 0,
30
- "POSITIVE": 1
31
- },
32
- "id2label": {
33
- "0": "NEGATIVE",
34
- "1": "POSITIVE"
35
- }
36
  }
1
  {
2
+ "activation": "gelu",
3
  "architectures": [
4
  "DistilBertForSequenceClassification"
5
  ],
 
6
  "attention_dropout": 0.1,
7
  "dim": 768,
8
  "dropout": 0.1,
9
  "finetuning_task": "sst-2",
10
  "hidden_dim": 3072,
11
+ "id2label": {
12
+ "0": "NEGATIVE",
13
+ "1": "POSITIVE"
14
+ },
15
  "initializer_range": 0.02,
16
+ "label2id": {
17
+ "NEGATIVE": 0,
18
+ "POSITIVE": 1
19
+ },
20
  "max_position_embeddings": 512,
21
+ "model_type": "distilbert",
22
  "n_heads": 12,
23
  "n_layers": 6,
 
 
 
24
  "output_past": true,
25
+ "pad_token_id": 0,
26
  "qa_dropout": 0.1,
27
  "seq_classif_dropout": 0.2,
28
  "sinusoidal_pos_embds": false,
29
  "tie_weights_": true,
30
+ "vocab_size": 30522
 
 
 
 
 
 
 
 
 
 
31
  }