diff --git a/funasr/bin/train.py b/funasr/bin/train.py index 15180710b..ccb65dc34 100755 --- a/funasr/bin/train.py +++ b/funasr/bin/train.py @@ -388,8 +388,8 @@ def get_parser(): if __name__ == '__main__': parser = get_parser() - args = parser.parse_args() - task_args = build_args(args) + # args = parser.parse_args() + task_args = build_args(parser) args = argparse.Namespace(**vars(args), **vars(task_args)) # set random seed diff --git a/funasr/build_utils/build_args.py b/funasr/build_utils/build_args.py index 0ada43993..b77cbcc4c 100644 --- a/funasr/build_utils/build_args.py +++ b/funasr/build_utils/build_args.py @@ -85,5 +85,5 @@ def build_args(args): else: raise NotImplementedError("Not supported task: {}".format(args.task_name)) - args = parser.parse_args() - return args + task_args = parser.parse_args() + return task_args