/EfficientPyTorch

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

Primary LanguageJupyter Notebook

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