A small example package
Project description
Chinese [English](https://gitee.com/casia_iva_engineer/pruning-tools/blob/master/README_english.md) ## 概述 EasyPruner是一个轻量且实用的PyTorch神经网络剪枝工具包,提供了一系列即插即用的网络结构裁剪接口。任何不具备模型压缩知识背景的工程师都可通过在PyTorch工程代码中添加几行代码实现网络模型的精准瘦身,并导出用于多种平台部署的onnx模型。可使模型保持现有精度水平的情况下,成倍提升执行效率和存储效率。<br /><br /> <a name=”EHucI”></a> ## 特点 透明性:不需懂模型压缩知识即可流畅使用。<br />**灵活性**:无需更换训练框架,也不用为剪枝接口重构训练或评测代码,仅在原训练框架中进行几行代码增 加,即插即用。<br />**通用性**:支持所有工程场景中常用的网络结构,例如ResNet、VGGNet、Inception、MobileNet等;支持多种训练框架代码,open-mmlab系列开源框架、u版YOLO系列等。<br />**精确性**:吸取神经网络剪枝的最新研究成果,提供在公开评测集上处于SOTA水平的剪枝方法,可对网络冗余连接进行精确识别,在一些常规任务上可以实现无损压缩。<br />**实用性**:剪枝后的模型可直接导出onnx,实现在NPU、ARM、GPU、CPU等多种平台的通用部署<br />
<a name=”BXs1U”></a> ## 创新 与其他开源模型剪枝工具项目相比,本项目做了如下创新,以使剪枝工具更加强大:
提出基于ONNX的图分析的网络拓扑排序方法,可以实现前后依赖算子的自动化识别,以兼容多样化的训练框架。
提出包含自研GradDecay稀疏剪枝方法的进阶剪枝模式,该方法针对YOLOv5目标检测算法设计,可以在不增加训练成本的情况下,最大限度保留网络原始表达能力,在一般情况下实现无损剪枝,在大剪枝率情况下依旧保持低精度损失。
提出包含自研MaskL1稀疏剪枝方法的进阶剪枝模式,该方法比目前流行的Network Slimming 方法有优的剪枝精度。
详细使用文档请参加语雀:
中文版: https://www.yuque.com/books/share/d1639c26-4a93-4274-b028-3134ebcada17?# 《EasyPruner剪枝工具》
英文版: https://www.yuque.com/docs/share/2abac59c-942d-413c-a379-e28fef97288f
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file easypruner-1.0.0.tar.gz.
File metadata
- Download URL: easypruner-1.0.0.tar.gz
- Upload date:
- Size: 30.3 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/3.4.2 importlib_metadata/3.10.0 pkginfo/1.7.0 requests/2.25.1 requests-toolbelt/0.9.1 tqdm/4.59.0 CPython/3.8.8
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
b8a82b1b759e4c05164f75906a766efb5b5d4527a79d4069ffa10ab0191e098b
|
|
| MD5 |
5f39c5f63e87510fb4367bbdec8645ec
|
|
| BLAKE2b-256 |
98b27f8074bd2ce0750850c5c553ff78517f1a474665f95b477fff38bda595f6
|
File details
Details for the file easypruner-1.0.0-py3-none-any.whl.
File metadata
- Download URL: easypruner-1.0.0-py3-none-any.whl
- Upload date:
- Size: 36.0 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/3.4.2 importlib_metadata/3.10.0 pkginfo/1.7.0 requests/2.25.1 requests-toolbelt/0.9.1 tqdm/4.59.0 CPython/3.8.8
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
8975b4d5e6a15a07a5e0fd7b65c05055641ded1a606b2e06e7b612dd53df97f0
|
|
| MD5 |
eb13e490381d06f95aa0b0e28f6c1889
|
|
| BLAKE2b-256 |
a4e31ecebe60c30339fd600ac4144385dd38f55da78f7a9b7eb5aada2ef42375
|