from local

This commit is contained in:
dohe0342 2023-02-25 16:10:30 +09:00
parent e173f765e8
commit ca5e95929f
2 changed files with 1 additions and 0 deletions

View File

@ -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)