Pruning Project for EfficientNet
It supports two different implementations of EfficientNet in https://github.com/lukemelas/EfficientNet-PyTorch and https://github.com/zylo117/Yet-Another-EfficientDet-Pytorch
- git clone https://github.com/voidbag/EfficientNet-Pruning-PyTorch.git
- git submodule init
- git submodule update
- ept: EfficientNet-PyTorch
- yet: Yet-Another-EfficientDet-Pytorch
./test_ept.sh
./test_yet.sh