Are there any PyTorch libraries, that can serve to prune networks?
Tensorflow 2.0 is arrives with built in functionality for pruning: https://www.tensorflow.org/model_optimization/guide/pruning/pruning_with_keras
There are some out-of-the-box pruning functionalities in PyTorch too now, starting from v1.4. You can find them under torch.nn.utils.prune.
1 Like