From 4897f2c0f2c18735b3282bdfa0926b621efe8f52 Mon Sep 17 00:00:00 2001
From: JinZr <60612200+JinZr@users.noreply.github.com>
Date: Thu, 28 Sep 2023 11:38:28 +0800
Subject: [PATCH] minor updates

---
 egs/multi_zh_en/ASR/zipformer/decode.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/egs/multi_zh_en/ASR/zipformer/decode.py b/egs/multi_zh_en/ASR/zipformer/decode.py
index 337e5a540..dce1d9a37 100755
--- a/egs/multi_zh_en/ASR/zipformer/decode.py
+++ b/egs/multi_zh_en/ASR/zipformer/decode.py
@@ -304,6 +304,13 @@ def get_parser():
         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)
 
     return parser
@@ -788,7 +795,7 @@ def main():
     # we need cut ids to display recognition results.
     args.return_cuts = True
     data_module = AsrDataModule(args)
-    multi_dataset = MultiDataset(args.manifest_dir)
+    multi_dataset = MultiDataset(args)
 
     def remove_short_utt(c: Cut):
         T = ((c.num_frames - 7) // 2 + 1) // 2