from local

This commit is contained in:
dohe0342 2023-02-25 19:38:45 +09:00
parent 082d326d65
commit cde3f3ca84
2 changed files with 1 additions and 1 deletions

Binary file not shown.

View File

@ -980,7 +980,7 @@ def rescore_with_attention_decoder(
tot_scores = (
am_scores.values
+ n_scale * ngram_lm_scores.values
a_scale * attention_scores
+ a_scale * attention_scores
)
ragged_tot_scores = k2.RaggedTensor(nbest.shape, tot_scores)
max_indexes = ragged_tot_scores.argmax()