mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-11 06:55:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

from local

Browse Source
This commit is contained in:
dohe0342 2023-02-25 15:38:05 +09:00
parent 48b2dddcd9
commit ef329c07b5
2 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
egs/librispeech/ASR/conformer_ctc2/.decode.py.swp
View File

Binary file not shown.

13
egs/librispeech/ASR/conformer_ctc2/decode.py
View File

@ -411,6 +411,19 @@ def decode_one_batch(
key = "ctc-greedy-search"
return {key: hyps}
if params.method == "greedy-search":
hyps = greedy_search(nnet_output, memory_key_padding_mask)
# hyps is a list of str, e.g., ['xxx yyy zzz', ...]
hyps = bpe_model.decode(hyps)
# hyps is a list of list of str, e.g., [['xxx', 'yyy', 'zzz'], ... ]
unk = bpe_model.decode(bpe_model.unk_id()).strip()
hyps = [[w for w in s.split() if w != unk] for s in hyps]
key = "ctc-greedy-search"
return {key: hyps}
if params.method == "nbest-oracle":
# Note: You can also pass rescored lattices to it.