update inference params
Browse files
README.md
CHANGED
@@ -59,12 +59,12 @@ Nearby tokens should be important because, in a sentence (sequence of words), th
|
|
59 |
inference:
|
60 |
parameters:
|
61 |
max_length: 64
|
62 |
-
min_length:
|
63 |
-
no_repeat_ngram_size:
|
64 |
early_stopping: True
|
65 |
-
repetition_penalty:
|
66 |
-
length_penalty: 0.
|
67 |
-
encoder_no_repeat_ngram_size :
|
68 |
num_beams : 4
|
69 |
|
70 |
|
|
|
59 |
inference:
|
60 |
parameters:
|
61 |
max_length: 64
|
62 |
+
min_length: 8
|
63 |
+
no_repeat_ngram_size: 3
|
64 |
early_stopping: True
|
65 |
+
repetition_penalty: 3.5
|
66 |
+
length_penalty: 0.1
|
67 |
+
encoder_no_repeat_ngram_size : 4
|
68 |
num_beams : 4
|
69 |
|
70 |
|