johngiorgi
commited on
Commit
•
7be7569
1
Parent(s):
8d70caf
Update config.json
Browse filesAdd `num_beams: 5` and `no_repeat_ngram_size: 3` to the config as defaults. Like `allenai/PRIMERA-multinews` and `allenai/PRIMERA-multixscience`, this are required to get scores closer to the scores reported in the original PRIMERA paper.
- config.json +2 -0
config.json
CHANGED
@@ -73,6 +73,8 @@
|
|
73 |
"model_type": "led",
|
74 |
"normalize_before": false,
|
75 |
"normalize_embedding": true,
|
|
|
|
|
76 |
"num_hidden_layers": 12,
|
77 |
"output_past": false,
|
78 |
"pad_token_id": 1,
|
|
|
73 |
"model_type": "led",
|
74 |
"normalize_before": false,
|
75 |
"normalize_embedding": true,
|
76 |
+
"num_beams": 5,
|
77 |
+
"no_repeat_ngram_size": 3,
|
78 |
"num_hidden_layers": 12,
|
79 |
"output_past": false,
|
80 |
"pad_token_id": 1,
|