from local

This commit is contained in:
dohe0342 2023-02-25 18:09:24 +09:00
parent 17300ba995
commit 041af601da
2 changed files with 4 additions and 3 deletions

Binary file not shown.

View File

@ -967,11 +967,12 @@ def rescore_with_attention_decoder(
ngram_lm_scale_list = [ngram_lm_scale]
if attention_scale is None:
attention_scale_list = [1.0]
#attention_scale_list = [0.01, 0.05, 0.08]
#attention_scale_list += [0.1, 0.3, 0.5, 0.6, 0.7, 0.9, 1.0]
attention_scale_list = [1.1, 1.2, 1.3, 1.5, 1.7, 1.9, 2.0]
attention_scale_list += [2.1, 2.2, 2.3, 2.5, 3.0, 4.0, 5.0]
attention_scale_list += [6.0, 7.0, 8.0, 9.0, 10.0, 20.0, 30.0, 40.0]
#attention_scale_list = [1.1, 1.2, 1.3, 1.5, 1.7, 1.9, 2.0]
#attention_scale_list += [2.1, 2.2, 2.3, 2.5, 3.0, 4.0, 5.0]
#attention_scale_list += [6.0, 7.0, 8.0, 9.0, 10.0, 20.0, 30.0, 40.0]
else:
attention_scale_list = [attention_scale]