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

Fix an error.

Browse Source 
Nbest.fsa should always have token IDs as labels and
word IDs as aux_labels.
This commit is contained in:
Fangjun Kuang 2021-09-13 20:55:26 +08:00
parent 9ddf23636e
commit 968e4a6609
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
icefall/decode2.py
View File

@ -202,7 +202,8 @@ class Nbest(object):
path_lattice, use_double_scores=use_double_scores path_lattice, use_double_scores=use_double_scores
) )
one_best = k2.remove_epsilon(one_best) one_best = k2.invert(one_best)
# Now one_best has token IDs as labels and word IDs as aux_labels
return Nbest(fsa=one_best, shape=self.shape) return Nbest(fsa=one_best, shape=self.shape)