from local
This commit is contained in:
parent
89945a0e11
commit
f4abe493d2
Binary file not shown.
@ -427,7 +427,7 @@ def decode_one_batch(
|
|||||||
return_output=True,
|
return_output=True,
|
||||||
)
|
)
|
||||||
print(pred.size())
|
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 is a list of str, e.g., ['xxx yyy zzz', ...]
|
||||||
hyps = bpe_model.decode(hyps)
|
hyps = bpe_model.decode(hyps)
|
||||||
|
|||||||
Reference in New Issue
Block a user