嘉渊
2023-04-24 e86b95e747286d2854864156b5edd12062467b3f
update
3个文件已修改
8 ■■■■ 已修改文件
funasr/bin/train.py 5 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/build_utils/build_args.py 1 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/utils/prepare_data.py 2 ●●● 补丁 | 查看 | 原始文档 | blame | 历史
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)
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
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)