def METHOD_NAME():
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "--config", default="../configs/transformer.big.yaml", type=str, help="Path of the config file. "
    )
    parser.add_argument(
        "--device", default="gpu", type=str, choices=["gpu", "cpu"], help="Device to use during inference. "
    )
    parser.add_argument("--model_dir", default="", type=str, help="Path of the model. ")
    parser.add_argument(
        "--benchmark",
        action="store_true",
        help="Whether to print logs on each cards and use benchmark vocab. Normally, not necessary to set --benchmark. ",
    )
    parser.add_argument("--profile", action="store_true", help="Whether to profile. ")
    args = parser.METHOD_NAME()
    return args