yashita13 commited on
Commit
f758a94
·
verified ·
1 Parent(s): 7041588

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +17 -15
config.json CHANGED
@@ -1,13 +1,14 @@
1
  {
2
- "activation": "gelu",
3
  "architectures": [
4
- "DistilBertForSequenceClassification"
5
  ],
6
- "attention_dropout": 0.1,
7
- "dim": 768,
8
- "dropout": 0.1,
9
  "dtype": "float32",
10
- "hidden_dim": 3072,
 
 
 
11
  "id2label": {
12
  "0": "LABEL_0",
13
  "1": "LABEL_1",
@@ -15,22 +16,23 @@
15
  "3": "LABEL_3"
16
  },
17
  "initializer_range": 0.02,
 
18
  "label2id": {
19
  "LABEL_0": 0,
20
  "LABEL_1": 1,
21
  "LABEL_2": 2,
22
  "LABEL_3": 3
23
  },
 
24
  "max_position_embeddings": 512,
25
- "model_type": "distilbert",
26
- "n_heads": 12,
27
- "n_layers": 6,
28
  "pad_token_id": 0,
 
29
  "problem_type": "single_label_classification",
30
- "qa_dropout": 0.1,
31
- "seq_classif_dropout": 0.2,
32
- "sinusoidal_pos_embds": false,
33
- "tie_weights_": true,
34
- "transformers_version": "4.56.1",
35
  "vocab_size": 30522
36
- }
 
1
  {
 
2
  "architectures": [
3
+ "BertForSequenceClassification"
4
  ],
5
+ "attention_probs_dropout_prob": 0.1,
6
+ "classifier_dropout": null,
 
7
  "dtype": "float32",
8
+ "gradient_checkpointing": false,
9
+ "hidden_act": "gelu",
10
+ "hidden_dropout_prob": 0.1,
11
+ "hidden_size": 768,
12
  "id2label": {
13
  "0": "LABEL_0",
14
  "1": "LABEL_1",
 
16
  "3": "LABEL_3"
17
  },
18
  "initializer_range": 0.02,
19
+ "intermediate_size": 3072,
20
  "label2id": {
21
  "LABEL_0": 0,
22
  "LABEL_1": 1,
23
  "LABEL_2": 2,
24
  "LABEL_3": 3
25
  },
26
+ "layer_norm_eps": 1e-12,
27
  "max_position_embeddings": 512,
28
+ "model_type": "bert",
29
+ "num_attention_heads": 12,
30
+ "num_hidden_layers": 12,
31
  "pad_token_id": 0,
32
+ "position_embedding_type": "absolute",
33
  "problem_type": "single_label_classification",
34
+ "transformers_version": "4.56.2",
35
+ "type_vocab_size": 2,
36
+ "use_cache": true,
 
 
37
  "vocab_size": 30522
38
+ }