correct params
Browse files- config.json +5 -15
config.json
CHANGED
@@ -32,28 +32,18 @@
|
|
32 |
"LABEL_1": 1,
|
33 |
"LABEL_2": 2
|
34 |
},
|
35 |
-
"length_penalty":
|
36 |
-
"max_length":
|
37 |
-
"max_position_embeddings":
|
38 |
-
"min_length":
|
39 |
"model_type": "bart",
|
40 |
"no_repeat_ngram_size": 3,
|
41 |
"normalize_before": false,
|
42 |
-
"num_beams":
|
43 |
"num_hidden_layers": 12,
|
44 |
"output_past": true,
|
45 |
"pad_token_id": 1,
|
46 |
"prefix": " ",
|
47 |
"scale_embedding": false,
|
48 |
-
"task_specific_params": {
|
49 |
-
"summarization": {
|
50 |
-
"early_stopping": true,
|
51 |
-
"length_penalty": 0.6,
|
52 |
-
"max_length": 32,
|
53 |
-
"min_length": 0,
|
54 |
-
"no_repeat_ngram_size": 3,
|
55 |
-
"num_beams": 8
|
56 |
-
}
|
57 |
-
},
|
58 |
"vocab_size": 50264
|
59 |
}
|
|
|
32 |
"LABEL_1": 1,
|
33 |
"LABEL_2": 2
|
34 |
},
|
35 |
+
"length_penalty": 0.6,
|
36 |
+
"max_length": 32,
|
37 |
+
"max_position_embeddings": 128,
|
38 |
+
"min_length": 0,
|
39 |
"model_type": "bart",
|
40 |
"no_repeat_ngram_size": 3,
|
41 |
"normalize_before": false,
|
42 |
+
"num_beams": 8,
|
43 |
"num_hidden_layers": 12,
|
44 |
"output_past": true,
|
45 |
"pad_token_id": 1,
|
46 |
"prefix": " ",
|
47 |
"scale_embedding": false,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
48 |
"vocab_size": 50264
|
49 |
}
|