from local

This commit is contained in:
dohe0342 2023-02-25 16:24:05 +09:00
parent a31218ada2
commit e67de22873
2 changed files with 1 additions and 3 deletions

View File

@ -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', ...]