This project implementation of the channel pruning through BN networks slimming in Pytorch.
|---dataset
|---dataset.py
|---models
|---model.py
|---new_resnet.py
|---resnet_bn_slim.py
|---config.py
|---train.py
|---utils.py
|---real_prune.py
CUB-200-2011 is a bird classi cation task with 11,788 images from 200 wild bird species. The ratio of train data and test data is roughly 1 : 1. It is generally considered one of the most competitive datasets since each species has only 30 images for training.
requires python3.6, pytorch 0.40
pip install -r requirements.txt
python train.py
- Fine tune.