From 52f910aae2d6427ab0bec40d257f47d23a4a263a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=98=89=E6=B8=8A?= <wangjiaming.wjm@alibaba-inc.com>
Date: Tue, 25 Apr 2023 01:02:28 +0800
Subject: [PATCH] update

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

diff --git a/funasr/bin/train.py b/funasr/bin/train.py
index 916693c1c..d3ebaacbf 100755
--- a/funasr/bin/train.py
+++ b/funasr/bin/train.py
@@ -297,6 +297,12 @@ def get_parser():
         default="small",
         help="whether to use dataloader for large dataset",
     )
+    parser.add_argument(
+        "--dataset_conf",
+        action=NestedDictAction,
+        default=dict(),
+        help=f"The keyword arguments for dataset",
+    )
     parser.add_argument(
         "--train_data_file",
         type=str,