From 9847fd24a81404b7491f35972bc7c14bf85aaedd Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=98=89=E6=B8=8A?= <wangjiaming.wjm@alibaba-inc.com>
Date: Mon, 24 Apr 2023 19:10:24 +0800
Subject: [PATCH] update

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

diff --git a/funasr/bin/train.py b/funasr/bin/train.py
index 7a9986cf4..26e0e6a26 100755
--- a/funasr/bin/train.py
+++ b/funasr/bin/train.py
@@ -284,6 +284,17 @@ def get_parser():
         action="append",
         default=[],
     )
+    parser.add_argument(
+        "--train_shape_file",
+        type=str, action="append",
+        default=[],
+    )
+    parser.add_argument(
+        "--valid_shape_file",
+        type=str,
+        action="append",
+        default=[],
+    )
     parser.add_argument(
         "--use_preprocessor",
         type=str2bool,