commit | fea1516f94cfcbd801124e3fdc4b5f5c4526e15b | [log] [tgz] |
---|---|---|
author | Rickard Bolin <rickard.bolin@arm.com> | Mon Jul 04 16:19:16 2022 +0000 |
committer | Rickard Bolin <rickard.bolin@arm.com> | Fri Sep 23 09:13:20 2022 +0000 |
tree | 92b991244ef535d652d0bb6e875e9e3a289257f5 | |
parent | cc219be4ec175645e8457da80d5effbf4324943b [diff] |
MLBEDSW-6686: Resize bilinear HPC with tile padding - Added support for Resize Bilinear with half pixel centers for int8 and uint8. - Utilizes the new "TILE" padding mode. - Utilizes ofm stride multipliers and modified tile base offsets to write OFMs interleaved. Signed-off-by: Rickard Bolin <rickard.bolin@arm.com> Change-Id: I37fa77c022a368f05fda0ead75d8696c9205f833
This tool is used to compile a TensorFlow Lite for Microcontrollers neural network model into an optimised version that can run on an embedded system containing an Arm Ethos-U NPU.
In order to be accelerated by the Ethos-U NPU the network operators must be quantised to either 8-bit (unsigned or signed) or 16-bit (signed).
The optimised model will contain TensorFlow Lite Custom operators for those parts of the model that can be accelerated by the Ethos-U NPU. Parts of the model that cannot be accelerated are left unchanged and will instead run on the Cortex-M series CPU using an appropriate kernel (such as the Arm optimised CMSIS-NN kernels).
After compilation the optimised model can only be run on an Ethos-U NPU embedded system.
The tool will also generate performance estimates (EXPERIMENTAL) for the compiled model.
The tool has limited functionality for compiling a TOSA neural network (EXPERIMENTAL).
Vela runs on Linux and Microsoft Windows 10 operating systems.
The following should be installed prior to the installation of Vela:
apt install python3.7-dev
or yum install python37-devel
Vela is available to install as a package from PyPi, or as source code from ML Platform. Both methods will automatically install all the required dependencies.
Install Vela from PyPi using the following command:
pip3 install ethos-u-vela
First obtain the source code by either downloading the desired TGZ file from:
https://review.mlplatform.org/plugins/gitiles/ml/ethos-u/ethos-u-vela
Or by cloning the git repository:
git clone https://review.mlplatform.org/ml/ethos-u/ethos-u-vela.git
Once you have the source code, Vela can be installed using the following command from the root directory of the repository:
pip3 install .
If you plan to modify the Vela codebase then it is recommended to install Vela as an editable package to avoid the need to re-install after every modification. This is done by adding the -e
option to the install command like so:
pip3 install -e .
If you plan to contribute to the Vela project (highly encouraged!) then it is recommended to install Vela along with the pre-commit tools (see Vela Testing for more details).
Vela is run with an input .tflite
or .tosa
(EXPERIMENTAL) file passed on the command line. This file contains the neural network to be compiled. The tool then outputs an optimised .tflite
file with a _vela
suffix in the file name, along with performance estimate (EXPERIMENTAL) CSV files, all to the output directory. It also prints a performance estimation summary back to the console, see Vela Performance Estimation Summary.
Example usage:
my_model.tflite
. The optimised version will be output to ./output/my_network_vela.tflite
.vela my_model.tflite
/path/to/my_model.tflite
and specify the output to go in the directory ./results_dir/
.vela --output-dir ./results_dir /path/to/my_model.tflite
vela --accelerator-config ethos-u65-512 my_model.tflite
vela --optimise Size my_model.tflite
vela --optimise Performance my_model.tflite
vela --optimise Performance --arena-cache-size 300000 my_model.tflite
My_Sys_Config
system configuration along with the My_Mem_Mode
memory mode from the vela.ini
configuration file located in the config_files directory.vela --config Arm/vela.ini --system-config My_Sys_Config --memory-mode My_Mem_Mode my_model.tflite
vela --list-config-files
vela --help
When running the Vela compiler it may report a number of warning messages to the console. These should all be thoroughly reviewed as they will indicate decisions that the compiler has made in order to create the optimised network.
Some example networks that contain quantised operators which can be compiled by Vela to run on the Ethos-U NPU can be found at: https://tfhub.dev/s?deployment-format=lite&q=quantized
Please see Vela External APIs.
Please see Vela Contributions.
Please see Vela Debug Database.
Please see Vela CLI Options. This includes a description of the system configuration file format.
Please see Vela Performance Estimation Summary.
Please see Vela Releases.
Please see Vela Security.
Please see Vela Supported Operators for the list of operators supported in this release.
Please see Vela Testing.
Please see Vela Community Bug Reporting for a description of how to report bugs.
Additional useful information:
Vela is licensed under Apache License 2.0.