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
ca5e95929f
commit
ee845f219c
Binary file not shown.
@ -417,7 +417,7 @@ def decode_one_batch(
|
|||||||
|
|
||||||
if params.method == "greedy-search":
|
if params.method == "greedy-search":
|
||||||
unsorted_token_ids = graph_compiler.texts_to_ids(supervisions["text"])
|
unsorted_token_ids = graph_compiler.texts_to_ids(supervisions["text"])
|
||||||
att_loss = model.decoder_forward(
|
pred, att_loss = model.decoder_forward(
|
||||||
memory,
|
memory,
|
||||||
memory_key_padding_mask,
|
memory_key_padding_mask,
|
||||||
token_ids=unsorted_token_ids,
|
token_ids=unsorted_token_ids,
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user