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
3d1ff70da4
commit
a66a1fcae0
Binary file not shown.
@ -415,7 +415,7 @@ def decode_one_batch(
|
|||||||
return {key: hyps}
|
return {key: hyps}
|
||||||
|
|
||||||
if params.method == "greedy-search":
|
if params.method == "greedy-search":
|
||||||
att_loss = mmodel.decoder_forward(
|
att_loss = model.decoder_forward(
|
||||||
encoder_memory,
|
encoder_memory,
|
||||||
memory_mask,
|
memory_mask,
|
||||||
token_ids=unsorted_token_ids,
|
token_ids=unsorted_token_ids,
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user