Pytorch implementation of "Learnable Lookup Table for Neural Network Quantization", CVPR 2022
- PyTorch
- TorchVision
- numpy
python train.py --arch resnet20 --epochs 200 --batch_size 128 --learning_rate 0.01 --weight_decay 1e-4 --w_bits 4 --a_bits 4
python test.py --arch resnet20 --batch_size 128 --w_bits 4 --a_bits 4
To be updated
To be updated
@InProceedings{Wang2022Learnable,
author = {Wang, Longguang and Dong, Xiaoyu and Wang, Yingqian and Liu, Li and An, Wei and Guo, Yulan},
title = {Learnable Lookup Table for Neural Network Quantization},
booktitle = {CVPR},
year = {2022},
pages = {12423--12433},
}
Part of the code is borrowed from APot. We thank the authors for sharing the codes.