from local

This commit is contained in:
dohe0342 2023-02-25 16:11:01 +09:00
parent 89945a0e11
commit f4abe493d2
2 changed files with 1 additions and 1 deletions

View File

@ -427,7 +427,7 @@ def decode_one_batch(
return_output=True,
)
print(pred.size())
hyps = greedy_search(nnet_output, memory_key_padding_mask)
hyps = greedy_search(pred, memory_key_padding_mask)
# hyps is a list of str, e.g., ['xxx yyy zzz', ...]
hyps = bpe_model.decode(hyps)