mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
from local
This commit is contained in:
parent
2e348bb1c9
commit
7eb4c3a619
Binary file not shown.
Binary file not shown.
@ -954,7 +954,7 @@ def rescore_with_attention_decoder(
|
|||||||
assert nll.shape[0] == len(token_ids)
|
assert nll.shape[0] == len(token_ids)
|
||||||
|
|
||||||
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.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 += [0.1, 0.3, 0.5, 0.6, 0.7, 0.9, 1.0]
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user