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
e173f765e8
commit
ca5e95929f
Binary file not shown.
@ -424,6 +424,7 @@ def decode_one_batch(
|
|||||||
sos_id=graph_compiler.sos_id,
|
sos_id=graph_compiler.sos_id,
|
||||||
eos_id=graph_compiler.eos_id,
|
eos_id=graph_compiler.eos_id,
|
||||||
warmup=warmup,
|
warmup=warmup,
|
||||||
|
return_output=True,
|
||||||
)
|
)
|
||||||
hyps = greedy_search(nnet_output, memory_key_padding_mask)
|
hyps = greedy_search(nnet_output, memory_key_padding_mask)
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user