PENNI

This repo provides the code of PENNI: Pruned Kernel Sharing for Efficient CNN Inference.

Install Requirement Packages:

pip install -r requirements.txt

If you find this work is helpful, cite with:

@inproceedings{li2020penni,
title={PENNI: Pruned Kernel Sharing for Efficient CNN Inference},
author={Li, Shiyu and Hanson, Edward and Li, Hai and Chen, Yiran},
booktitle={International Conference on Machine Learning},
year={2020}
}

Acknowledgement

The resnet-56 implementation is from: pytorch_resnet_cifar10

We count the FLOPs number with the modified version of pytorch-OpCounter

The ImageNet training script is derived from apex