Björn Barz commited on
Commit
6c5cfdf
1 Parent(s): f3b1f9f

Use sampling by default instead of beam search

Browse files
Files changed (1) hide show
  1. config.json +5 -2
config.json CHANGED
@@ -151,8 +151,11 @@
151
  "max_length": 32,
152
  "min_length": 3,
153
  "model_type": "encoder-decoder",
154
- "no_repeat_ngram_size": 3,
155
- "num_beams": 4,
 
 
 
156
  "pad_token_id": 0,
157
  "vocab_size": 30522
158
  }
151
  "max_length": 32,
152
  "min_length": 3,
153
  "model_type": "encoder-decoder",
154
+ "no_repeat_ngram_size": 2,
155
+ "num_beams": 10,
156
+ "do_sample": true,
157
+ "temperature": 1.5,
158
+ "top_k": 50,
159
  "pad_token_id": 0,
160
  "vocab_size": 30522
161
  }