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

---
 .../pruned_transducer_stateless7_streaming/train2.py   | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/egs/aishell/ASR/pruned_transducer_stateless7_streaming/train2.py b/egs/aishell/ASR/pruned_transducer_stateless7_streaming/train2.py
index 5692edf83..88eb34104 100755
--- a/egs/aishell/ASR/pruned_transducer_stateless7_streaming/train2.py
+++ b/egs/aishell/ASR/pruned_transducer_stateless7_streaming/train2.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."
     )