hustzxd / EfficientPyTorch

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

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

• Learned Step Quantization (LSQ)

References