johngiorgi commited on
Commit
7be7569
1 Parent(s): 8d70caf

Update config.json

Browse files

Add `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.

Files changed (1) hide show
  1. 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,