def METHOD_NAME():
    parser = argparse.ArgumentParser()
    parser.add_argument("--backend", type=str, default="tcp", help="Name of the backend to use.")
    parser.add_argument(
        "--init-method",
        "-i",
        type=str,
        default=os.environ.get("INIT_METHOD", "tcp://127.0.0.1:23456"),
        help="URL specifying how to initialize the package.",
    )
    parser.add_argument(
        "--rank",
        "-r",
        type=int,
        default=int(os.environ.get("RANK")),
        help="Rank of the current process.",
    )
    parser.add_argument(
        "--world-size",
        "-s",
        type=int,
        default=int(os.environ.get("WORLD")),
        help="Number of processes participating in the job.",
    )
    parser.add_argument("--epochs", type=int, default=20)
    parser.add_argument("--no-cuda", action="store_true")
    parser.add_argument("--learning-rate", "-lr", type=float, default=1e-3)
    parser.add_argument("--root", type=str, default="data")
    parser.add_argument("--batch-size", type=int, default=128)
    args = parser.parse_args()
    args.cuda = torch.cuda.is_available() and not args.no_cuda
    print(args)
    init_process(args)
    solve(args)