Tobi-Bueck commited on
Commit
4588956
·
verified ·
1 Parent(s): 6835680

Upload BertForSequenceClassification

Browse files
Files changed (2) hide show
  1. config.json +14 -14
  2. model.safetensors +2 -2
config.json CHANGED
@@ -1,12 +1,12 @@
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",
@@ -20,6 +20,7 @@
20
  "9": "LABEL_9"
21
  },
22
  "initializer_range": 0.02,
 
23
  "label2id": {
24
  "LABEL_0": 0,
25
  "LABEL_1": 1,
@@ -32,18 +33,17 @@
32
  "LABEL_8": 8,
33
  "LABEL_9": 9
34
  },
 
35
  "max_position_embeddings": 512,
36
- "model_type": "distilbert",
37
- "n_heads": 12,
38
- "n_layers": 6,
39
- "output_past": true,
40
  "pad_token_id": 0,
 
41
  "problem_type": "single_label_classification",
42
- "qa_dropout": 0.1,
43
- "seq_classif_dropout": 0.2,
44
- "sinusoidal_pos_embds": true,
45
- "tie_weights_": true,
46
  "torch_dtype": "float32",
47
  "transformers_version": "4.52.4",
 
 
48
  "vocab_size": 31102
49
  }
 
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",
 
20
  "9": "LABEL_9"
21
  },
22
  "initializer_range": 0.02,
23
+ "intermediate_size": 3072,
24
  "label2id": {
25
  "LABEL_0": 0,
26
  "LABEL_1": 1,
 
33
  "LABEL_8": 8,
34
  "LABEL_9": 9
35
  },
36
+ "layer_norm_eps": 1e-12,
37
  "max_position_embeddings": 512,
38
+ "model_type": "bert",
39
+ "num_attention_heads": 12,
40
+ "num_hidden_layers": 12,
 
41
  "pad_token_id": 0,
42
+ "position_embedding_type": "absolute",
43
  "problem_type": "single_label_classification",
 
 
 
 
44
  "torch_dtype": "float32",
45
  "transformers_version": "4.52.4",
46
+ "type_vocab_size": 2,
47
+ "use_cache": true,
48
  "vocab_size": 31102
49
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:cce7a0a7b89fb931ead733a0f1285ec674d556c2294077d12c7315067f06f5a1
3
- size 269638936
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:416276223fa64f0f4ff45563bc3b80bfc5ba12f099efb3512286600691cee91c
3
+ size 439765016