mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-10 10:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

liear_fst_with_self_loops (#512)

Browse Source
This commit is contained in:
LIyong.Guo 2022-08-02 22:28:12 +08:00 committed by GitHub
parent 2f75236c05
commit 132132f52a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
icefall/decode.py
View File

@ -330,12 +330,14 @@ class Nbest(object):
# We use a word fsa to intersect with k2.invert(lattice) # We use a word fsa to intersect with k2.invert(lattice)
word_fsa = k2.invert(self.fsa) word_fsa = k2.invert(self.fsa)
word_fsa.scores.zero_()
if hasattr(lattice, "aux_labels"): if hasattr(lattice, "aux_labels"):
# delete token IDs as it is not needed # delete token IDs as it is not needed
del word_fsa.aux_labels del word_fsa.aux_labels
word_fsa_with_epsilon_loops = k2.linear_fsa_with_self_loops(word_fsa)
else:
word_fsa_with_epsilon_loops = k2.linear_fst_with_self_loops(word_fsa)
word_fsa.scores.zero_()
word_fsa_with_epsilon_loops = k2.linear_fsa_with_self_loops(word_fsa)
path_to_utt_map = self.shape.row_ids(1) path_to_utt_map = self.shape.row_ids(1)