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 Data preparation
Read more