system HF staff commited on
Commit
74c5919
1 Parent(s): 3c11da4

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +7 -20
config.json CHANGED
@@ -6,43 +6,30 @@
6
  "attention_dropout": 0.1,
7
  "bos_token_id": 0,
8
  "dim": 768,
9
- "do_sample": false,
10
  "dropout": 0.1,
11
  "eos_token_ids": 0,
12
  "finetuning_task": "mnli",
13
  "hidden_dim": 3072,
14
  "id2label": {
15
- "0": "LABEL_0",
16
- "1": "LABEL_1"
 
17
  },
18
  "initializer_range": 0.02,
19
- "is_decoder": false,
20
  "label2id": {
21
- "LABEL_0": 0,
22
- "LABEL_1": 1
 
23
  },
24
- "length_penalty": 1.0,
25
- "max_length": 20,
26
  "max_position_embeddings": 512,
 
27
  "n_heads": 12,
28
  "n_layers": 6,
29
- "num_beams": 1,
30
- "num_labels": 3,
31
- "num_return_sequences": 1,
32
- "output_attentions": false,
33
- "output_hidden_states": false,
34
  "output_past": true,
35
  "pad_token_id": 0,
36
- "pruned_heads": {},
37
  "qa_dropout": 0.1,
38
- "repetition_penalty": 1.0,
39
  "seq_classif_dropout": 0.2,
40
  "sinusoidal_pos_embds": false,
41
- "temperature": 1.0,
42
  "tie_weights_": true,
43
- "top_k": 50,
44
- "top_p": 1.0,
45
- "torchscript": false,
46
- "use_bfloat16": false,
47
  "vocab_size": 30522
48
  }
 
6
  "attention_dropout": 0.1,
7
  "bos_token_id": 0,
8
  "dim": 768,
 
9
  "dropout": 0.1,
10
  "eos_token_ids": 0,
11
  "finetuning_task": "mnli",
12
  "hidden_dim": 3072,
13
  "id2label": {
14
+ "0": "contradiction",
15
+ "1": "neutral",
16
+ "2": "entailment"
17
  },
18
  "initializer_range": 0.02,
 
19
  "label2id": {
20
+ "contradiction": "0",
21
+ "neutral": "1",
22
+ "entailment": "2"
23
  },
 
 
24
  "max_position_embeddings": 512,
25
+ "model_type": "distilbert",
26
  "n_heads": 12,
27
  "n_layers": 6,
 
 
 
 
 
28
  "output_past": true,
29
  "pad_token_id": 0,
 
30
  "qa_dropout": 0.1,
 
31
  "seq_classif_dropout": 0.2,
32
  "sinusoidal_pos_embds": false,
 
33
  "tie_weights_": true,
 
 
 
 
34
  "vocab_size": 30522
35
  }