def METHOD_NAME(extra_args):
    parser = argparse.ArgumentParser()
    parser.add_argument("--task_name", default="cola", choices=task_to_keys.keys(), help="Name of task to run")
    # validate in train by default
    parser.add_argument("--validate_in_train", action="store_false", help="Validate result in train")
    # use fp16 mixed precision by default
    parser.add_argument("--fp16", default="amp", choices=["amp", "no"], help="Enable mixed precision")
    parser.add_argument(
        "--distributed", default="none", choices=["ddp", "fsdp", "deepspeed", "none"],
        help="distributed training paradigm, by default using DDP"
    )
    tb_args = parser.parse_args(extra_args)
    return tb_args