from local

This commit is contained in:
dohe0342 2023-02-25 18:00:03 +09:00
parent fc1d3b5d8a
commit 51137a579d
3 changed files with 3 additions and 0 deletions

Binary file not shown.

View File

@ -956,10 +956,13 @@ def rescore_with_attention_decoder(
attention_scores = -nll.sum(dim=1)
if ngram_lm_scale is None:
ngram_lm_scale_list = [0.]
'''
ngram_lm_scale_list = [0.01, 0.05, 0.08]
ngram_lm_scale_list += [0.1, 0.3, 0.5, 0.6, 0.7, 0.9, 1.0]
ngram_lm_scale_list += [1.1, 1.2, 1.3, 1.5, 1.7, 1.9, 2.0]
ngram_lm_scale_list += [2.1, 2.2, 2.3, 2.5, 3.0, 4.0, 5.0]
'''
else:
ngram_lm_scale_list = [ngram_lm_scale]