def METHOD_NAME():
    parser = argparse.ArgumentParser(description='Model training')
    parser.add_argument("--config", help="The path of config file.", type=str)
    parser.add_argument(
        "--input_shape",
        nargs='+',
        type=int,
        help="Set the input shape, such as --input_shape 1 3 1024 1024",
        default=[1, 3, 1024, 1024])
    return parser.METHOD_NAME()