pszemraj commited on
Commit
2e4030b
1 Parent(s): b6289aa

update inference params

Browse files
Files changed (1) hide show
  1. README.md +5 -5
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: 4
63
- no_repeat_ngram_size: 2
64
  early_stopping: True
65
- repetition_penalty: 2.4
66
- length_penalty: 0.3
67
- encoder_no_repeat_ngram_size : 3
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