system HF staff commited on
Commit
6fd4954
1 Parent(s): 949bc3d

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +12 -3
config.json CHANGED
@@ -1,7 +1,9 @@
1
  {
 
2
  "activation_dropout": 0.0,
3
  "architectures": null,
4
  "attention_dropout": 0.0,
 
5
  "classif_dropout": 0.0,
6
  "d_model": 1024,
7
  "decoder_attention_heads": 16,
@@ -10,10 +12,14 @@
10
  "decoder_layers": 12,
11
  "do_sample": false,
12
  "dropout": 0.1,
 
13
  "encoder_attention_heads": 16,
14
  "encoder_ffn_dim": 4096,
15
  "encoder_layerdrop": 0.0,
16
  "encoder_layers": 12,
 
 
 
17
  "finetuning_task": null,
18
  "id2label": {
19
  "0": "LABEL_0",
@@ -22,22 +28,25 @@
22
  },
23
  "init_std": 0.02,
24
  "is_decoder": false,
 
25
  "label2id": {
26
  "LABEL_0": 0,
27
  "LABEL_1": 1,
28
  "LABEL_2": 2
29
  },
30
- "length_penalty": 1.0,
31
  "max_length": 20,
32
  "max_position_embeddings": 1024,
 
33
  "model_type": "bart",
34
- "num_beams": 1,
 
35
  "num_hidden_layers": 12,
36
- "num_labels": 3,
37
  "num_return_sequences": 1,
38
  "output_attentions": false,
39
  "output_hidden_states": false,
40
  "output_past": true,
 
41
  "pruned_heads": {},
42
  "repetition_penalty": 1.0,
43
  "temperature": 1.0,
 
1
  {
2
+ "_num_labels": 3,
3
  "activation_dropout": 0.0,
4
  "architectures": null,
5
  "attention_dropout": 0.0,
6
+ "bos_token_id": 0,
7
  "classif_dropout": 0.0,
8
  "d_model": 1024,
9
  "decoder_attention_heads": 16,
 
12
  "decoder_layers": 12,
13
  "do_sample": false,
14
  "dropout": 0.1,
15
+ "early_stopping": false,
16
  "encoder_attention_heads": 16,
17
  "encoder_ffn_dim": 4096,
18
  "encoder_layerdrop": 0.0,
19
  "encoder_layers": 12,
20
+ "eos_token_ids": [
21
+ 2
22
+ ],
23
  "finetuning_task": null,
24
  "id2label": {
25
  "0": "LABEL_0",
 
28
  },
29
  "init_std": 0.02,
30
  "is_decoder": false,
31
+ "is_encoder_decoder": true,
32
  "label2id": {
33
  "LABEL_0": 0,
34
  "LABEL_1": 1,
35
  "LABEL_2": 2
36
  },
37
+ "length_penalty": 2.0,
38
  "max_length": 20,
39
  "max_position_embeddings": 1024,
40
+ "min_length": 0,
41
  "model_type": "bart",
42
+ "no_repeat_ngram_size": 3,
43
+ "num_beams": 4,
44
  "num_hidden_layers": 12,
 
45
  "num_return_sequences": 1,
46
  "output_attentions": false,
47
  "output_hidden_states": false,
48
  "output_past": true,
49
+ "pad_token_id": 1,
50
  "pruned_heads": {},
51
  "repetition_penalty": 1.0,
52
  "temperature": 1.0,