qat

Step-by-Step

This document is used to list steps of reproducing Intel® Neural Compressor QAT feature.

Prerequisite

1. Installation

# Install Intel® Neural Compressor
pip install neural-compressor

2. Install Intel Tensorflow 2.4.0 or above.

pip install intel-tensorflow==2.4.0

3. Install tensorflow_model_optimization

pip install tensorflow_model_optimization

Run Command

python resnet_v2.py    # to get the quantized ResNet-V2 model which will be saved into './trained_qat_model'.