Salvatore commited on
Commit
960bf64
·
verified ·
1 Parent(s): 8c1d5eb

Upload BertForSequenceClassification

Browse files
Files changed (2) hide show
  1. config.json +25 -13
  2. model.safetensors +2 -2
config.json CHANGED
@@ -1,23 +1,35 @@
1
  {
2
- "activation": "gelu",
3
  "architectures": [
4
- "DistilBertForSequenceClassification"
5
  ],
6
- "attention_dropout": 0.1,
7
- "dim": 768,
8
- "dropout": 0.1,
9
- "hidden_dim": 3072,
 
 
 
 
 
 
 
10
  "initializer_range": 0.02,
 
 
 
 
 
 
 
11
  "max_position_embeddings": 512,
12
- "model_type": "distilbert",
13
- "n_heads": 12,
14
- "n_layers": 6,
15
  "pad_token_id": 0,
16
- "qa_dropout": 0.1,
17
- "seq_classif_dropout": 0.2,
18
- "sinusoidal_pos_embds": false,
19
- "tie_weights_": true,
20
  "torch_dtype": "float32",
21
  "transformers_version": "4.51.3",
 
 
22
  "vocab_size": 30522
23
  }
 
1
  {
 
2
  "architectures": [
3
+ "BertForSequenceClassification"
4
  ],
5
+ "attention_probs_dropout_prob": 0.1,
6
+ "classifier_dropout": null,
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
+ },
16
  "initializer_range": 0.02,
17
+ "intermediate_size": 3072,
18
+ "label2id": {
19
+ "LABEL_0": 0,
20
+ "LABEL_1": 1,
21
+ "LABEL_2": 2
22
+ },
23
+ "layer_norm_eps": 1e-12,
24
  "max_position_embeddings": 512,
25
+ "model_type": "bert",
26
+ "num_attention_heads": 12,
27
+ "num_hidden_layers": 12,
28
  "pad_token_id": 0,
29
+ "position_embedding_type": "absolute",
 
 
 
30
  "torch_dtype": "float32",
31
  "transformers_version": "4.51.3",
32
+ "type_vocab_size": 2,
33
+ "use_cache": true,
34
  "vocab_size": 30522
35
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:cc25d7c9c58f81ff99dd6b94cf536fcc234e95ae3ffef28296b4f0216c7939ed
3
- size 267832560
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c5f30873c38ff36dafca1a12169175ece8371828e217cacc6cab891301b3c254
3
+ size 437961724