Björn Barz
commited on
Commit
•
6c5cfdf
1
Parent(s):
f3b1f9f
Use sampling by default instead of beam search
Browse files- 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":
|
155 |
-
"num_beams":
|
|
|
|
|
|
|
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 |
}
|