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

minor fix

Browse Source
This commit is contained in:
Dongji Gao 2023-09-25 15:05:42 -04:00
parent 3da93d942a
commit 8a3051bea7
1 changed files with 0 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
icefall/otc_graph_compiler.py
View File

@ -121,7 +121,6 @@ class OtcTrainingGraphCompiler(object):
allow_self_loop_arc: str2bool = True, allow_self_loop_arc: str2bool = True,
bypass_weight: float = 0.0, bypass_weight: float = 0.0,
self_loop_weight: float = 0.0, self_loop_weight: float = 0.0,
otc_granularity: str = "word",
) -> k2.Fsa: ) -> k2.Fsa:
"""Build a OTC graph from a texts (list of words). """Build a OTC graph from a texts (list of words).
@ -141,14 +140,11 @@ class OtcTrainingGraphCompiler(object):
Weight associated with bypass arc. Weight associated with bypass arc.
self_loop_weight: self_loop_weight:
Weight associated with self-loop arc. Weight associated with self-loop arc.
otc_granularity:
Use OTC token to model word or subword.
Return: Return:
Return an FsaVec, which is the result of composing a Return an FsaVec, which is the result of composing a
CTC topology with OTC FSAs constructed from the given texts. CTC topology with OTC FSAs constructed from the given texts.
""" """
assert otc_granularity in ("word", "subword")
transcript_fsa = self.convert_transcript_to_fsa( transcript_fsa = self.convert_transcript_to_fsa(
texts, texts,
@ -157,7 +153,6 @@ class OtcTrainingGraphCompiler(object):
allow_self_loop_arc, allow_self_loop_arc,
bypass_weight, bypass_weight,
self_loop_weight, self_loop_weight,
otc_granularity,
) )
transcript_fsa = transcript_fsa.to(self.device) transcript_fsa = transcript_fsa.to(self.device)
fsa_with_self_loop = k2.remove_epsilon_and_add_self_loops(transcript_fsa) fsa_with_self_loop = k2.remove_epsilon_and_add_self_loops(transcript_fsa)