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

minor updates

Browse Source
This commit is contained in:
JinZr 2023-09-28 11:38:28 +08:00
parent 8f2d480439
commit 4897f2c0f2
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
egs/multi_zh_en/ASR/zipformer/decode.py
View File

@ -304,6 +304,13 @@ def get_parser():
fast_beam_search_nbest_LG, and fast_beam_search_nbest_oracle""", fast_beam_search_nbest_LG, and fast_beam_search_nbest_oracle""",
) )
parser.add_argument(
"--use-tal-csasr",
type=str2bool,
default=False,
help="Whether to use TAL-CSASR training data.",
)
add_model_arguments(parser) add_model_arguments(parser)
return parser return parser
@ -788,7 +795,7 @@ def main():
# we need cut ids to display recognition results. # we need cut ids to display recognition results.
args.return_cuts = True args.return_cuts = True
data_module = AsrDataModule(args) data_module = AsrDataModule(args)
multi_dataset = MultiDataset(args.manifest_dir) multi_dataset = MultiDataset(args)
def remove_short_utt(c: Cut): def remove_short_utt(c: Cut):
T = ((c.num_frames - 7) // 2 + 1) // 2 T = ((c.num_frames - 7) // 2 + 1) // 2