This commit is contained in:
嘉渊 2023-04-24 16:22:50 +08:00
parent ea0118a4cb
commit d56b8c2bb5
2 changed files with 4 additions and 4 deletions

View File

@ -388,8 +388,8 @@ def get_parser():
if __name__ == '__main__': if __name__ == '__main__':
parser = get_parser() parser = get_parser()
args = parser.parse_args() # args = parser.parse_args()
task_args = build_args(args) task_args = build_args(parser)
args = argparse.Namespace(**vars(args), **vars(task_args)) args = argparse.Namespace(**vars(args), **vars(task_args))
# set random seed # set random seed

View File

@ -85,5 +85,5 @@ def build_args(args):
else: else:
raise NotImplementedError("Not supported task: {}".format(args.task_name)) raise NotImplementedError("Not supported task: {}".format(args.task_name))
args = parser.parse_args() task_args = parser.parse_args()
return args return task_args