system HF staff commited on
Commit
5cb3aa4
1 Parent(s): 28db7ad

Update config.json

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