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,