Skip to content

datuizhuang/channel_prune

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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)

About

channel prune for resnet

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages