model_compression:PyTorch模型压缩

时间:2024-05-12 06:22:01
【文件属性】:

文件名称:model_compression:PyTorch模型压缩

文件大小:91KB

文件格式:ZIP

更新时间:2024-05-12 06:22:01

pytorch pruning quantization lottey-ticket-hypothesis Python

模型压缩 内容 非结构化修剪 结构化修剪 结构化修剪后收缩 量化 类图 参考 贡献者 入门 先决条件 该存储库是在Python3.7的Anaconda虚拟环境上实现和验证的 安装 克隆此存储库。 $ git clone https://github.com/j-marple-dev/model_compression.git $ cd model_compression 创建虚拟环境 $ conda env create -f environment.yml $ conda activate model_compression 或者 $ make install $ conda activate model_compression (贡献者可选)安装CI环境 $ conda activate model_compression $ make dev (对于nvidia


【文件预览】:
model_compression-master
----shrink.py(1KB)
----quantize.py(2KB)
----mypy.ini(77B)
----.flake8(247B)
----train.py(2KB)
----prune.py(1KB)
----.github()
--------weekly-digest.yml(222B)
--------workflows()
----src()
--------models()
--------regularizers.py(1KB)
--------utils.py(4KB)
--------plotter.py(9KB)
--------runners()
--------lr_schedulers.py(3KB)
--------format.py(344B)
--------augmentation()
--------criterions.py(7KB)
----Dockerfile(1KB)
----.pylintrc(17KB)
----__init__.py(0B)
----.isort.cfg(172B)
----config()
--------quantize()
--------train()
--------prune()
--------pretrained_model_url.yaml(548B)
--------config_validator.py(14KB)
----LICENSE(1KB)
----environment.yml(447B)
----README.md(22KB)
----Makefile(579B)
----.pre-commit-config.yaml(240B)
----.all-contributorsrc(1KB)
----tests()
--------test_dummy.py(178B)
----.gitignore(2KB)

网友评论