├── .gitattributes ├── .gitignore ├── Neurips2019Poster.pdf ├── Neurips2019Presentation.pdf ├── README.md ├── __init__.py ├── bias_correction.ipynb ├── bit_allocation_synthetic.py ├── fig ├── aciq-mse.png ├── bias-corr-eq-12.png ├── bias-corr-eq-13.png ├── bin-alloc-eq10.png ├── bit-alloc-mse.png ├── bit_alloc-eq-11.png ├── experiments.png ├── opt_clipping-eq-6.png └── resnet101_bias_err.png ├── inference ├── __init__.py └── inference_sim.py ├── kernels ├── build_all.sh ├── build_int_quantization.py ├── gemmlowp.cu └── int_quantization.cpp ├── mse_analysis.py ├── pytorch_quantizer ├── __init__.py ├── clipping │ ├── __init__.py │ └── clipping_manager.py └── quantization │ ├── __init__.py │ ├── inference │ ├── __init__.py │ ├── angle_stats.py │ ├── distance_stats.py │ ├── inference_quantization_manager.py │ ├── kld_threshold.py │ ├── measure_statistics.py │ ├── statistic_manager.py │ └── statistic_manager_perchannel.py │ ├── kmeans_quantization.py │ ├── qtypes │ ├── __init__.py │ ├── dummy_quantizer.py │ └── int_quantizer.py │ └── quantization_manager.py └── utils ├── __init__.py ├── absorb_bn.py ├── attacher.py ├── dataset.py ├── dump_manager.py ├── entropy.py ├── log.py ├── mark_relu.py ├── meters.py ├── misc.py ├── mllog.py ├── model_naming.py ├── monitor.py ├── optim.py └── preprocess.py /.gitattributes: -------------------------------------------------------------------------------- 1 | jupyter/* linguist-detectable=false 2 | -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/.gitignore -------------------------------------------------------------------------------- /Neurips2019Poster.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/Neurips2019Poster.pdf -------------------------------------------------------------------------------- /Neurips2019Presentation.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/Neurips2019Presentation.pdf -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /bias_correction.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/bias_correction.ipynb -------------------------------------------------------------------------------- /bit_allocation_synthetic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/bit_allocation_synthetic.py -------------------------------------------------------------------------------- /fig/aciq-mse.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/fig/aciq-mse.png -------------------------------------------------------------------------------- /fig/bias-corr-eq-12.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/fig/bias-corr-eq-12.png -------------------------------------------------------------------------------- /fig/bias-corr-eq-13.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/fig/bias-corr-eq-13.png -------------------------------------------------------------------------------- /fig/bin-alloc-eq10.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/fig/bin-alloc-eq10.png -------------------------------------------------------------------------------- /fig/bit-alloc-mse.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/fig/bit-alloc-mse.png -------------------------------------------------------------------------------- /fig/bit_alloc-eq-11.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/fig/bit_alloc-eq-11.png -------------------------------------------------------------------------------- /fig/experiments.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/fig/experiments.png -------------------------------------------------------------------------------- /fig/opt_clipping-eq-6.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/fig/opt_clipping-eq-6.png -------------------------------------------------------------------------------- /fig/resnet101_bias_err.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/fig/resnet101_bias_err.png -------------------------------------------------------------------------------- /inference/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /inference/inference_sim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/inference/inference_sim.py -------------------------------------------------------------------------------- /kernels/build_all.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/kernels/build_all.sh -------------------------------------------------------------------------------- /kernels/build_int_quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/kernels/build_int_quantization.py -------------------------------------------------------------------------------- /kernels/gemmlowp.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/kernels/gemmlowp.cu -------------------------------------------------------------------------------- /kernels/int_quantization.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/kernels/int_quantization.cpp -------------------------------------------------------------------------------- /mse_analysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/mse_analysis.py -------------------------------------------------------------------------------- /pytorch_quantizer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pytorch_quantizer/clipping/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /pytorch_quantizer/clipping/clipping_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/pytorch_quantizer/clipping/clipping_manager.py -------------------------------------------------------------------------------- /pytorch_quantizer/quantization/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pytorch_quantizer/quantization/inference/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /pytorch_quantizer/quantization/inference/angle_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/pytorch_quantizer/quantization/inference/angle_stats.py -------------------------------------------------------------------------------- /pytorch_quantizer/quantization/inference/distance_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/pytorch_quantizer/quantization/inference/distance_stats.py -------------------------------------------------------------------------------- /pytorch_quantizer/quantization/inference/inference_quantization_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/pytorch_quantizer/quantization/inference/inference_quantization_manager.py -------------------------------------------------------------------------------- /pytorch_quantizer/quantization/inference/kld_threshold.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/pytorch_quantizer/quantization/inference/kld_threshold.py -------------------------------------------------------------------------------- /pytorch_quantizer/quantization/inference/measure_statistics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/pytorch_quantizer/quantization/inference/measure_statistics.py -------------------------------------------------------------------------------- /pytorch_quantizer/quantization/inference/statistic_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/pytorch_quantizer/quantization/inference/statistic_manager.py -------------------------------------------------------------------------------- /pytorch_quantizer/quantization/inference/statistic_manager_perchannel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/pytorch_quantizer/quantization/inference/statistic_manager_perchannel.py -------------------------------------------------------------------------------- /pytorch_quantizer/quantization/kmeans_quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/pytorch_quantizer/quantization/kmeans_quantization.py -------------------------------------------------------------------------------- /pytorch_quantizer/quantization/qtypes/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/pytorch_quantizer/quantization/qtypes/__init__.py -------------------------------------------------------------------------------- /pytorch_quantizer/quantization/qtypes/dummy_quantizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/pytorch_quantizer/quantization/qtypes/dummy_quantizer.py -------------------------------------------------------------------------------- /pytorch_quantizer/quantization/qtypes/int_quantizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/pytorch_quantizer/quantization/qtypes/int_quantizer.py -------------------------------------------------------------------------------- /pytorch_quantizer/quantization/quantization_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/pytorch_quantizer/quantization/quantization_manager.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/absorb_bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/utils/absorb_bn.py -------------------------------------------------------------------------------- /utils/attacher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/utils/attacher.py -------------------------------------------------------------------------------- /utils/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/utils/dataset.py -------------------------------------------------------------------------------- /utils/dump_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/utils/dump_manager.py -------------------------------------------------------------------------------- /utils/entropy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/utils/entropy.py -------------------------------------------------------------------------------- /utils/log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/utils/log.py -------------------------------------------------------------------------------- /utils/mark_relu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/utils/mark_relu.py -------------------------------------------------------------------------------- /utils/meters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/utils/meters.py -------------------------------------------------------------------------------- /utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/utils/misc.py -------------------------------------------------------------------------------- /utils/mllog.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/utils/mllog.py -------------------------------------------------------------------------------- /utils/model_naming.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/utils/model_naming.py -------------------------------------------------------------------------------- /utils/monitor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/utils/monitor.py -------------------------------------------------------------------------------- /utils/optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/utils/optim.py -------------------------------------------------------------------------------- /utils/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/submission2019/cnn-quantization/HEAD/utils/preprocess.py --------------------------------------------------------------------------------