【文件属性】:
文件名称:model_compression:PyTorch模型压缩
文件大小:91KB
文件格式:ZIP
更新时间:2021-04-22 12:35:21
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)