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
0b5398db0c
commit
a29a52b500
Binary file not shown.
Binary file not shown.
@ -954,6 +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)
|
||||||
|
attention_scores.size())
|
||||||
|
|
||||||
if ngram_lm_scale is None:
|
if ngram_lm_scale is None:
|
||||||
ngram_lm_scale_list = [0.]
|
ngram_lm_scale_list = [0.]
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user