Skip to main content

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> ## 创新 与其他开源模型剪枝工具项目相比,本项目做了如下创新,以使剪枝工具更加强大:

  1. 提出基于ONNX的图分析的网络拓扑排序方法,可以实现前后依赖算子的自动化识别,以兼容多样化的训练框架。

  2. 提出包含自研GradDecay稀疏剪枝方法的进阶剪枝模式,该方法针对YOLOv5目标检测算法设计,可以在不增加训练成本的情况下,最大限度保留网络原始表达能力,在一般情况下实现无损剪枝,在大剪枝率情况下依旧保持低精度损失。

  3. 提出包含自研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


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

easypruner-1.0.0.tar.gz (30.3 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

easypruner-1.0.0-py3-none-any.whl (36.0 kB view details)

Uploaded Python 3

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

Hashes for easypruner-1.0.0.tar.gz
Algorithm Hash digest
SHA256 b8a82b1b759e4c05164f75906a766efb5b5d4527a79d4069ffa10ab0191e098b
MD5 5f39c5f63e87510fb4367bbdec8645ec
BLAKE2b-256 98b27f8074bd2ce0750850c5c553ff78517f1a474665f95b477fff38bda595f6

See more details on using hashes here.

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

Hashes for easypruner-1.0.0-py3-none-any.whl
Algorithm Hash digest
SHA256 8975b4d5e6a15a07a5e0fd7b65c05055641ded1a606b2e06e7b612dd53df97f0
MD5 eb13e490381d06f95aa0b0e28f6c1889
BLAKE2b-256 a4e31ecebe60c30339fd600ac4144385dd38f55da78f7a9b7eb5aada2ef42375

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page