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

minor change

Browse Source
This commit is contained in:
yaozengwei 2023-02-07 21:42:02 +08:00
parent cca3113817
commit fa740be117
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
icefall/utils.py
View File

@ -1502,7 +1502,7 @@ def filter_uneven_sized_batch(batch: dict, allowed_max_frames: int):
def parse_bpe_start_end_pairs( def parse_bpe_start_end_pairs(
tokens: List[str], is_first_token: List[bool] tokens: List[str], is_first_token: List[bool]
) -> List[Tuple[int, int]]: ) -> List[Tuple[int, int]]:
"""Parse pairs of start and end frame indexes for each word, """Parse pairs of start and end frame indexes for each word.
Args: Args:
tokens: tokens:
@ -1571,7 +1571,7 @@ def parse_bpe_start_end_pairs(
def parse_bpe_timestamps_and_texts( def parse_bpe_timestamps_and_texts(
best_paths: k2.Fsa, sp: spm.SentencePieceProcessor best_paths: k2.Fsa, sp: spm.SentencePieceProcessor
) -> Tuple[List[Tuple[int, int]], List[List[str]]]: ) -> Tuple[List[Tuple[int, int]], List[List[str]]]:
"""Parse timestamps (frame indexes) and texts for BPE-based models. """Parse timestamps (frame indexes) and texts.
Args: Args:
best_paths: best_paths: