from local

This commit is contained in:
dohe0342 2023-02-26 23:20:59 +09:00
parent 738180141f
commit c7b4a3ba71
2 changed files with 5 additions and 4 deletions

Binary file not shown.

View File

@ -956,10 +956,11 @@ def rescore_with_attention_decoder(
attention_scores = -nll.sum(dim=1) attention_scores = -nll.sum(dim=1)
if ngram_lm_scale is None: if ngram_lm_scale is None:
ngram_lm_scale_list = [0.01, 0.05, 0.08] ngram_lm_scale_list = [0]
ngram_lm_scale_list += [0.1, 0.3, 0.5, 0.6, 0.7, 0.9, 1.0] #ngram_lm_scale_list = [0.01, 0.05, 0.08]
ngram_lm_scale_list += [1.1, 1.2, 1.3, 1.5, 1.7, 1.9, 2.0] #ngram_lm_scale_list += [0.1, 0.3, 0.5, 0.6, 0.7, 0.9, 1.0]
ngram_lm_scale_list += [2.1, 2.2, 2.3, 2.5, 3.0, 4.0, 5.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: else:
ngram_lm_scale_list = [ngram_lm_scale] ngram_lm_scale_list = [ngram_lm_scale]