From e86b95e747286d2854864156b5edd12062467b3f Mon Sep 17 00:00:00 2001
From: 嘉渊 <wangjiaming.wjm@alibaba-inc.com>
Date: 星期一, 24 四月 2023 22:57:04 +0800
Subject: [PATCH] update
---
funasr/utils/prepare_data.py | 2 +-
funasr/bin/train.py | 5 ++---
funasr/build_utils/build_args.py | 1 +
3 files changed, 4 insertions(+), 4 deletions(-)
diff --git a/funasr/bin/train.py b/funasr/bin/train.py
index d5e7089..8f40e24 100755
--- a/funasr/bin/train.py
+++ b/funasr/bin/train.py
@@ -462,10 +462,9 @@
if __name__ == '__main__':
parser = get_parser()
- args, extra_task_params = parser.parse_known_args()
+ common_args, extra_task_params = parser.parse_known_args()
if extra_task_params:
- args = build_args(args, parser, extra_task_params)
- # args = argparse.Namespace(**vars(args), **vars(task_args))
+ args = build_args(common_args, parser, extra_task_params)
# set random seed
set_all_random_seed(args.seed)
diff --git a/funasr/build_utils/build_args.py b/funasr/build_utils/build_args.py
index b19030d..517c85b 100644
--- a/funasr/build_utils/build_args.py
+++ b/funasr/build_utils/build_args.py
@@ -88,5 +88,6 @@
if not any(action.dest == a.dest for a in task_parser._actions):
task_parser._add_action(action)
+ task_parser.set_defaults(**vars(args))
task_args = task_parser.parse_args(extra_task_params)
return task_args
diff --git a/funasr/utils/prepare_data.py b/funasr/utils/prepare_data.py
index a9beaf4..8d02e04 100644
--- a/funasr/utils/prepare_data.py
+++ b/funasr/utils/prepare_data.py
@@ -160,7 +160,7 @@
if args.dataset_type == "small" and args.train_data_path_and_name_and_type is not None:
return
if args.dataset_type == "large" and args.train_data_file is not None:
- return
+ return
distributed = distributed_option.distributed
if not distributed or distributed_option.dist_rank == 0:
filter_wav_text(args.data_dir, args.train_set)
--
Gitblit v1.9.1