from local

This commit is contained in:
dohe0342 2023-02-25 14:33:10 +09:00
parent 0d3fb7d3f5
commit 3058e82f4f
2 changed files with 5 additions and 4 deletions

View File

@ -431,10 +431,11 @@ def decode_one_batch(
"whole-lattice-rescoring", "whole-lattice-rescoring",
"attention-decoder", "attention-decoder",
] ]
lm_scale_list = [0.01, 0.05, 0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7] lm_scale_list = [i/20 for i in range(20)]
lm_scale_list += [0.8, 0.9, 1.0, 1.1, 1.2, 1.3] #lm_scale_list = [0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7]
lm_scale_list += [1.4, 1.5, 1.6, 1.7, 1.8, 1.9, 2.0] #lm_scale_list += [0.8, 0.9, 1.0, 1.1, 1.2, 1.3]
#lm_scale_list += [1.4, 1.5, 1.6, 1.7, 1.8, 1.9, 2.0]
if params.method == "1best": if params.method == "1best":
best_path_dict = one_best_decoding( best_path_dict = one_best_decoding(