torch_pruning 剪枝

torch_pruning是一个基于PyTorch的剪枝库，用于在神经网络中剪枝不必要的连接，以减少模型的复杂度和计算量，同时提高模型的泛化能力。它提供了多种剪枝算法，包括全局剪枝、结构化剪枝、通道剪枝等，并且支持多种压缩技术，如权值共享、低秩分解等。除此之外，torch_pruning还提供了一些实用工具，如可视化剪枝效果、剪枝后的模型推理等，方便用户进行模型剪枝和优化。