GWCS contains three steps:
- train an original network.
- search pruned network by GGS algorithm.
- the pruned network KD from the original network to improving the accuracy.
-
Requirements: python3.6.5; pytorch 1.1.0; torchvision 0.4.0
-
Datasets: Cifar10, Cifarf100, ImageNet
-
Steps to run: train.py: train an original network search.py: search pruned network KD.py: improving the accuracy of pruning network
python train.py python search.py python KD.py