Hi, there is a nice repository for model compression in PyTorch here: https://github.com/opencv/openvino_training_extensions/tree/develop/pytorch_toolkit/nncf 215. Now it has two pruning methods and quantization support.