文件名称：musco-pytorch:穆斯科
文件大小：33KB
文件格式：ZIP
更新时间：2024-05-30 12:57:23
deep-neural-networks pytorch tensor-decomposition cp-decomposition tucker
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
【文件预览】：
musco-pytorch-master
----setup.py(1009B)
----.gitignore(160B)
----requirements.txt(69B)
----MANIFEST.in(41B)
----LICENSE(2KB)
----README.md(4KB)
----musco()
--------__init__.py(0B)
--------pytorch()