from local
This commit is contained in:
parent
082d326d65
commit
cde3f3ca84
Binary file not shown.
@ -980,7 +980,7 @@ def rescore_with_attention_decoder(
|
|||||||
tot_scores = (
|
tot_scores = (
|
||||||
am_scores.values
|
am_scores.values
|
||||||
+ n_scale * ngram_lm_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)
|
ragged_tot_scores = k2.RaggedTensor(nbest.shape, tot_scores)
|
||||||
max_indexes = ragged_tot_scores.argmax()
|
max_indexes = ragged_tot_scores.argmax()
|
||||||
|
|||||||
Reference in New Issue
Block a user