From 7c37a8f9337db6eed1d030a51756766e5c407cff Mon Sep 17 00:00:00 2001
From: jinzr <60612200+JinZr@users.noreply.github.com>
Date: Wed, 11 Oct 2023 09:55:05 +0800
Subject: [PATCH] Update train.py

---
 .../pruned_transducer_stateless7_streaming/train.py   | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/egs/aishell/ASR/pruned_transducer_stateless7_streaming/train.py b/egs/aishell/ASR/pruned_transducer_stateless7_streaming/train.py
index bbb4a368b..2e1044658 100755
--- a/egs/aishell/ASR/pruned_transducer_stateless7_streaming/train.py
+++ b/egs/aishell/ASR/pruned_transducer_stateless7_streaming/train.py
@@ -257,6 +257,16 @@ def get_parser():
         """,
     )
 
+    parser.add_argument(
+        "--lang-dir",
+        type=str,
+        default="data/lang_char",
+        help="""The lang dir
+        It contains language related input files such as
+        "lexicon.txt"
+        """,
+    )
+
     parser.add_argument(
         "--base-lr", type=float, default=0.05, help="The base learning rate."
     )
@@ -1224,6 +1234,7 @@ def main():
     AishellAsrDataModule.add_arguments(parser)
     args = parser.parse_args()
     args.exp_dir = Path(args.exp_dir)
+    args.lang_dir = Path(args.lang_dir)
 
     world_size = args.world_size
     assert world_size >= 1