Sharpness-aware Quantization for Deep Neural Networks
This is the official repository for our paper: Sharpness-aware Quantization for Deep Neural Networks by Jing Liu, Jianfei Cai, and Bohan Zhuang.
Recent Update
2021.11.24: We release the source code of SAQ.
Setup the environments
- Clone the repository locally:
git clone https://github.com/zhuang-group/SAQ
- Install pytorch 1.8+, tensorboard and prettytable
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
pip install tensorboard
pip install prettytable