VincentC12 commited on
Commit
0d8b3b5
1 Parent(s): 683a1fb

Upload config.json

Browse files
Files changed (1) hide show
  1. config.json +2 -12
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "distilbert-base-uncased-finetuned-sst-2-english",
3
  "activation": "gelu",
4
  "architectures": [
5
  "DistilBertForSequenceClassification"
@@ -7,22 +7,12 @@
7
  "attention_dropout": 0.1,
8
  "dim": 768,
9
  "dropout": 0.1,
10
- "finetuning_task": "sst-2",
11
  "hidden_dim": 3072,
12
- "id2label": {
13
- "0": "NEGATIVE",
14
- "1": "POSITIVE"
15
- },
16
  "initializer_range": 0.02,
17
- "label2id": {
18
- "NEGATIVE": 0,
19
- "POSITIVE": 1
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,
@@ -30,6 +20,6 @@
30
  "sinusoidal_pos_embds": false,
31
  "tie_weights_": true,
32
  "torch_dtype": "float32",
33
- "transformers_version": "4.16.2",
34
  "vocab_size": 30522
35
  }
 
1
  {
2
+ "_name_or_path": "distilbert-base-uncased",
3
  "activation": "gelu",
4
  "architectures": [
5
  "DistilBertForSequenceClassification"
 
7
  "attention_dropout": 0.1,
8
  "dim": 768,
9
  "dropout": 0.1,
 
10
  "hidden_dim": 3072,
 
 
 
 
11
  "initializer_range": 0.02,
 
 
 
 
12
  "max_position_embeddings": 512,
13
  "model_type": "distilbert",
14
  "n_heads": 12,
15
  "n_layers": 6,
 
16
  "pad_token_id": 0,
17
  "problem_type": "single_label_classification",
18
  "qa_dropout": 0.1,
 
20
  "sinusoidal_pos_embds": false,
21
  "tie_weights_": true,
22
  "torch_dtype": "float32",
23
+ "transformers_version": "4.15.0",
24
  "vocab_size": 30522
25
  }