Skip to content

PENNI: Pruned Kernel Sharing for Efficient CNN Inference

License

Notifications You must be signed in to change notification settings

timlee0212/PENNI

Folders and files

NameName
Last commit message
Last commit date

Latest commit

9e99c72 · Nov 30, 2020

History

5 Commits
Jun 24, 2020
Aug 24, 2020
Aug 24, 2020
Jun 24, 2020
Jun 24, 2020
Aug 24, 2020
Nov 30, 2020
Jun 24, 2020

Repository files navigation

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

About

PENNI: Pruned Kernel Sharing for Efficient CNN Inference

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published