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":
|
||||
unsorted_token_ids = graph_compiler.texts_to_ids(supervisions["text"])
|
||||
att_loss = model.decoder_forward(
|
||||
pred, att_loss = model.decoder_forward(
|
||||
memory,
|
||||
memory_key_padding_mask,
|
||||
token_ids=unsorted_token_ids,
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user