Tobi-Bueck commited on
Commit
28dd63c
·
verified ·
1 Parent(s): b07cd3e

Upload BertForSequenceClassification

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