Skip to content

Latest commit

 

History

History
12 lines (8 loc) · 372 Bytes

readme.md

File metadata and controls

12 lines (8 loc) · 372 Bytes

channel prune for ResNet

you can use this repo to prune your model(only support Res style block, such as BottleNeck and BasicBlock)

use this command to train and test on CIFAR10:

python CIFAR_train.py

Note:
PruneTool can prune the model with residual, but OldPruneTool ignore the residual(for BottleNeck, OldPruneTool will only prune the first two conv)