├── .gitignore ├── .travis.yml ├── LICENSE ├── README.md ├── experiments ├── Attributions comparison (Max model).ipynb ├── CIFAR-10 - VGG16 - Layerwise robustness.ipynb ├── Pruning Untrained Networks.ipynb ├── __init__.py ├── data │ └── .gitkeep ├── models │ ├── cifar10.py │ ├── fmnist.py │ └── mnist.py └── utils │ ├── __init__.py │ ├── train.py │ └── utils.py ├── setup.py └── torchpruner ├── __init__.py ├── attributions ├── __init__.py ├── attributions.py └── methods │ ├── __init__.py │ ├── apoz.py │ ├── random.py │ ├── sensitivity.py │ ├── shapley_values.py │ ├── taylor.py │ └── weight_norm.py ├── pruner ├── __init__.py ├── opt_pruner.py └── pruner.py ├── tests ├── __init__.py ├── test_attributions.py └── test_pruner.py └── utils ├── __init__.py └── graph.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/marcoancona/TorchPruner/HEAD/.gitignore -------------------------------------------------------------------------------- /.travis.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/marcoancona/TorchPruner/HEAD/.travis.yml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/marcoancona/TorchPruner/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/marcoancona/TorchPruner/HEAD/README.md -------------------------------------------------------------------------------- /experiments/Attributions comparison (Max model).ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/marcoancona/TorchPruner/HEAD/experiments/Attributions comparison (Max model).ipynb -------------------------------------------------------------------------------- /experiments/CIFAR-10 - VGG16 - Layerwise robustness.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/marcoancona/TorchPruner/HEAD/experiments/CIFAR-10 - VGG16 - Layerwise robustness.ipynb -------------------------------------------------------------------------------- /experiments/Pruning Untrained Networks.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/marcoancona/TorchPruner/HEAD/experiments/Pruning Untrained Networks.ipynb -------------------------------------------------------------------------------- /experiments/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /experiments/data/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /experiments/models/cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/marcoancona/TorchPruner/HEAD/experiments/models/cifar10.py -------------------------------------------------------------------------------- /experiments/models/fmnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/marcoancona/TorchPruner/HEAD/experiments/models/fmnist.py -------------------------------------------------------------------------------- /experiments/models/mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/marcoancona/TorchPruner/HEAD/experiments/models/mnist.py -------------------------------------------------------------------------------- /experiments/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/marcoancona/TorchPruner/HEAD/experiments/utils/__init__.py -------------------------------------------------------------------------------- /experiments/utils/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/marcoancona/TorchPruner/HEAD/experiments/utils/train.py -------------------------------------------------------------------------------- /experiments/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/marcoancona/TorchPruner/HEAD/experiments/utils/utils.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/marcoancona/TorchPruner/HEAD/setup.py -------------------------------------------------------------------------------- /torchpruner/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /torchpruner/attributions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/marcoancona/TorchPruner/HEAD/torchpruner/attributions/__init__.py -------------------------------------------------------------------------------- /torchpruner/attributions/attributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/marcoancona/TorchPruner/HEAD/torchpruner/attributions/attributions.py -------------------------------------------------------------------------------- /torchpruner/attributions/methods/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchpruner/attributions/methods/apoz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/marcoancona/TorchPruner/HEAD/torchpruner/attributions/methods/apoz.py -------------------------------------------------------------------------------- /torchpruner/attributions/methods/random.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/marcoancona/TorchPruner/HEAD/torchpruner/attributions/methods/random.py -------------------------------------------------------------------------------- /torchpruner/attributions/methods/sensitivity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/marcoancona/TorchPruner/HEAD/torchpruner/attributions/methods/sensitivity.py -------------------------------------------------------------------------------- /torchpruner/attributions/methods/shapley_values.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/marcoancona/TorchPruner/HEAD/torchpruner/attributions/methods/shapley_values.py -------------------------------------------------------------------------------- /torchpruner/attributions/methods/taylor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/marcoancona/TorchPruner/HEAD/torchpruner/attributions/methods/taylor.py -------------------------------------------------------------------------------- /torchpruner/attributions/methods/weight_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/marcoancona/TorchPruner/HEAD/torchpruner/attributions/methods/weight_norm.py -------------------------------------------------------------------------------- /torchpruner/pruner/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/marcoancona/TorchPruner/HEAD/torchpruner/pruner/__init__.py -------------------------------------------------------------------------------- /torchpruner/pruner/opt_pruner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/marcoancona/TorchPruner/HEAD/torchpruner/pruner/opt_pruner.py -------------------------------------------------------------------------------- /torchpruner/pruner/pruner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/marcoancona/TorchPruner/HEAD/torchpruner/pruner/pruner.py -------------------------------------------------------------------------------- /torchpruner/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torchpruner/tests/test_attributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/marcoancona/TorchPruner/HEAD/torchpruner/tests/test_attributions.py -------------------------------------------------------------------------------- /torchpruner/tests/test_pruner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/marcoancona/TorchPruner/HEAD/torchpruner/tests/test_pruner.py -------------------------------------------------------------------------------- /torchpruner/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/marcoancona/TorchPruner/HEAD/torchpruner/utils/__init__.py -------------------------------------------------------------------------------- /torchpruner/utils/graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/marcoancona/TorchPruner/HEAD/torchpruner/utils/graph.py --------------------------------------------------------------------------------