AI Model Efficiency Toolkit User Guide¶
AI Model Efficiency Toolkit (AIMET) is a software toolkit that enables users to compress and quantize models.
AIMET provides optimizations that are designed to be used on trained models, so that the resulting compressed and/or quantized model can be further trained (also called fine-tuning) to recover any accuracy losses. AIMET currently supports both TensorFlow and PyTorch models.
The above picture shows a high-level view of the workflow when using AIMET. The user will start with a trained model in either the TensorFlow or PyTorch training framework. This trained model is passed to AIMET using APIs for compression and quantization that we will look at in detail. AIMET returns a compressed/quantized version of the model that the users can fine-tune (or train further for a small number of epochs) to recover lost accuracy. Users can then export via ONNX to an on-target runtime like Qualcomm® Neural Processing SDK.
AIMET supports two types of features
Model Compression: AIMET supports multiple model compression techniques that allow the user to take a trained model and remove redundancies, resulting in a smaller model that runs faster on target.
Model Quantization: AIMET can simulate behavior of quantized HW for a given trained model. This allows the user to fine-tune the model on this simulated quantized HW. AIMET also offers post-training quantization techniques which make the model better for quantization.
API Documentation and Usage Examples¶
Please visit the generated AIMET API documentation here
Examples Documentation:Browse and use the Jupyter Notebooks that provide examples of using AIMET features.