├── .gitignore ├── LICENSE.md ├── README.md ├── configs ├── imagenet_resnet101_prune100.json ├── imagenet_resnet101_prune40.json ├── imagenet_resnet101_prune50.json ├── imagenet_resnet101_prune50_group.json ├── imagenet_resnet101_prune55.json ├── imagenet_resnet101_prune75.json ├── imagenet_resnet50_prune100.json ├── imagenet_resnet50_prune56.json ├── imagenet_resnet50_prune72.json ├── imagenet_resnet50_prune81.json └── imagenet_resnet50_prune91.json ├── images └── resnet_result.png ├── layers └── gate_layer.py ├── logger.py ├── main.py ├── models ├── densenet_imagenet.py ├── lenet.py ├── preact_resnet.py ├── resnet.py └── vgg_bn.py ├── pruning_engine.py ├── requirements.txt └── utils ├── group_lasso_optimizer.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Taylor_pruning/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Taylor_pruning/HEAD/LICENSE.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Taylor_pruning/HEAD/README.md -------------------------------------------------------------------------------- /configs/imagenet_resnet101_prune100.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Taylor_pruning/HEAD/configs/imagenet_resnet101_prune100.json -------------------------------------------------------------------------------- /configs/imagenet_resnet101_prune40.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Taylor_pruning/HEAD/configs/imagenet_resnet101_prune40.json -------------------------------------------------------------------------------- /configs/imagenet_resnet101_prune50.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Taylor_pruning/HEAD/configs/imagenet_resnet101_prune50.json -------------------------------------------------------------------------------- /configs/imagenet_resnet101_prune50_group.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Taylor_pruning/HEAD/configs/imagenet_resnet101_prune50_group.json -------------------------------------------------------------------------------- /configs/imagenet_resnet101_prune55.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Taylor_pruning/HEAD/configs/imagenet_resnet101_prune55.json -------------------------------------------------------------------------------- /configs/imagenet_resnet101_prune75.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Taylor_pruning/HEAD/configs/imagenet_resnet101_prune75.json -------------------------------------------------------------------------------- /configs/imagenet_resnet50_prune100.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Taylor_pruning/HEAD/configs/imagenet_resnet50_prune100.json -------------------------------------------------------------------------------- /configs/imagenet_resnet50_prune56.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Taylor_pruning/HEAD/configs/imagenet_resnet50_prune56.json -------------------------------------------------------------------------------- /configs/imagenet_resnet50_prune72.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Taylor_pruning/HEAD/configs/imagenet_resnet50_prune72.json -------------------------------------------------------------------------------- /configs/imagenet_resnet50_prune81.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Taylor_pruning/HEAD/configs/imagenet_resnet50_prune81.json -------------------------------------------------------------------------------- /configs/imagenet_resnet50_prune91.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Taylor_pruning/HEAD/configs/imagenet_resnet50_prune91.json -------------------------------------------------------------------------------- /images/resnet_result.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Taylor_pruning/HEAD/images/resnet_result.png -------------------------------------------------------------------------------- /layers/gate_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Taylor_pruning/HEAD/layers/gate_layer.py -------------------------------------------------------------------------------- /logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Taylor_pruning/HEAD/logger.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Taylor_pruning/HEAD/main.py -------------------------------------------------------------------------------- /models/densenet_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Taylor_pruning/HEAD/models/densenet_imagenet.py -------------------------------------------------------------------------------- /models/lenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Taylor_pruning/HEAD/models/lenet.py -------------------------------------------------------------------------------- /models/preact_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Taylor_pruning/HEAD/models/preact_resnet.py -------------------------------------------------------------------------------- /models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Taylor_pruning/HEAD/models/resnet.py -------------------------------------------------------------------------------- /models/vgg_bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Taylor_pruning/HEAD/models/vgg_bn.py -------------------------------------------------------------------------------- /pruning_engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Taylor_pruning/HEAD/pruning_engine.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Taylor_pruning/HEAD/requirements.txt -------------------------------------------------------------------------------- /utils/group_lasso_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Taylor_pruning/HEAD/utils/group_lasso_optimizer.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/Taylor_pruning/HEAD/utils/utils.py --------------------------------------------------------------------------------