EfficientPyTorch icon indicating copy to clipboard operation
EfficientPyTorch copied to clipboard

A PyTorch Framework for Efficient Pruning and Quantization for specialized accelerators.

Build Status Documentation Status

EfficientPyTorch

Install

git clone https://github.com/hustzxd/EfficientPyTorch.git

cd EfficientPyTorch
# ./setup.sh
source setup.sh
pip install -r requirements.txt

Run

./run_cli.sh *.prototxt
  • [x] Learned Step Quantization (LSQ)

References