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
a31218ada2
commit
e67de22873
Binary file not shown.
@ -403,8 +403,6 @@ def decode_one_batch(
|
||||
return {key: hyps}
|
||||
|
||||
if params.method == "ctc-greedy-search":
|
||||
print(nnet_output.size())
|
||||
exit()
|
||||
hyps = ctc_greedy_search(nnet_output, memory_key_padding_mask)
|
||||
|
||||
# hyps is a list of str, e.g., ['xxx yyy zzz', ...]
|
||||
@ -427,7 +425,7 @@ def decode_one_batch(
|
||||
eos_id=graph_compiler.eos_id,
|
||||
return_output=True,
|
||||
)
|
||||
print(pred.size())
|
||||
|
||||
hyps = greedy_search(pred, memory_key_padding_mask)
|
||||
|
||||
# hyps is a list of str, e.g., ['xxx yyy zzz', ...]
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user