From aa1fb476284d3d3be7973b4e8d36d18470f191ad Mon Sep 17 00:00:00 2001 From: Fangjun Kuang Date: Sat, 6 Nov 2021 08:49:50 +0800 Subject: [PATCH] Use a larger range for ngram_lm_scale and attention_scale --- icefall/decode.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/icefall/decode.py b/icefall/decode.py index 42622fa70..d11920618 100644 --- a/icefall/decode.py +++ b/icefall/decode.py @@ -871,6 +871,7 @@ def rescore_with_attention_decoder( 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] @@ -878,6 +879,7 @@ def rescore_with_attention_decoder( 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] else: attention_scale_list = [attention_scale]