speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
嘉渊 2023-04-24 22:57:04 +08:00
parent f2b9780b29
commit e86b95e747
3 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
funasr/bin/train.py
View File

@ -462,10 +462,9 @@ def get_parser():
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)

1
funasr/build_utils/build_args.py
View File

@ -88,5 +88,6 @@ def build_args(args, parser, extra_task_params):
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

2
funasr/utils/prepare_data.py
View File

@ -160,7 +160,7 @@ def prepare_data(args, distributed_option):
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)