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)
|
||||
|
||||
attention_scores = -nll.sum(dim=1)
|
||||
attention_scores.size())
|
||||
|
||||
if ngram_lm_scale is None:
|
||||
ngram_lm_scale_list = [0.]
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user