| | |
| | | |
| | | 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) |
| | |
| | | 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 |