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