musco-pytorch:穆斯科

MUSCO：神经网络的多阶段压缩 该存储库包含用于论文补充代码。 它演示了如何使用权重张量的迭代张量分解来压缩具有卷积层和完全连接层的神经网络。 要求 numpy scipy scikit-tensor-py3 absl-py flopco-pytorch tensorly==0.4.5 pytorch 安装 pip install musco-pytorch 快速开始 from torchvision . models import resnet50 from flopco import FlopCo from musco . pytorch import CompressorVBMF , CompressorPR , CompressorManual model = resnet50 ( pretrained = True ) model_stats = FlopCo ( model