From d13bd061b28067d65c1565b6d7ba87e64b8cea63 Mon Sep 17 00:00:00 2001 From: Tim Hall Date: Thu, 12 Nov 2020 23:29:25 +0000 Subject: vela: Update tool description text - Updated and aligned the --help and setup.py descriptions Signed-off-by: Tim Hall Change-Id: I78c11b1b3dd51284b34d57a6caca45cd222b4678 --- ethosu/vela/vela.py | 9 +++++++-- setup.py | 2 +- 2 files changed, 8 insertions(+), 3 deletions(-) diff --git a/ethosu/vela/vela.py b/ethosu/vela/vela.py index 05bd9ec1..6fc5c47f 100644 --- a/ethosu/vela/vela.py +++ b/ethosu/vela/vela.py @@ -187,7 +187,7 @@ def main(args=None): if args is None: args = sys.argv[1:] - parser = argparse.ArgumentParser(prog="vela", description="Neural network model compiler for Ethos-U") + parser = argparse.ArgumentParser(prog="vela", description="Neural network model compiler for Arm Ethos-U NPUs") parser.add_argument("--version", action="version", version=__version__) parser.add_argument( "--api-version", action="version", version=api_version, help="Displays the version of the external API." @@ -200,7 +200,12 @@ def main(args=None): # set network nargs to be optional to allow the support-ops-report CLI option to be used standalone parser.add_argument( - "network", metavar="NETWORK", type=str, default=None, nargs="?", help="Filename of network to process" + "network", + metavar="NETWORK", + type=str, + default=None, + nargs="?", + help="Filename of the input TensorFlow Lite for Microcontrollers network", ) parser.add_argument( "--output-dir", type=str, default="output", help="Output directory to write files to (default: %(default)s)" diff --git a/setup.py b/setup.py index 1ab8dd50..d3a1013e 100644 --- a/setup.py +++ b/setup.py @@ -34,7 +34,7 @@ mlw_module = Extension( setup( name="ethos-u-vela", use_scm_version=True, - description="Optimise TensorFlow Lite models for Ethos-U NPU.", + description="Neural network model compiler for Arm Ethos-U NPUs", long_description=long_description, long_description_content_type="text/markdown", url="https://git.mlplatform.org/ml/ethos-u/ethos-u-vela.git/", -- cgit v1.2.1