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