microsoft/nni

Does nni contain some new pruning base methods？

Open

#4,567 创建于 2022年2月21日

在 GitHub 查看
 (6 评论) (0 反应) (1 负责人)Python (13,504 star) (1,830 fork)batch import
good first issuemodel compressionnew feature

描述

Describe the issue: Sorry to bother you, in nni, the current basic pruning strategies are Norm, FPGM, ActivationPruner, TaylorFOWeightPruner, etc. Are there other basic strategies, such as HRank, if I want to add it myself, where do I need to rewrite it?

Environment:

  • NNI version: V2.6
  • Training service (local|remote|pai|aml|etc): local
  • Client OS: linux
  • Server OS (for remote mode only):
  • Python version: 3.8
  • PyTorch/TensorFlow version: pytorch1.8
  • Is conda/virtualenv/venv used?: conda
  • Is running in Docker?: no

How to reproduce it?:

贡献者指南

技术栈
pythonpytorch
领域
machine learningtooling
议题类型
research
难度面向新贡献者的预计实现难度，1 表示很小改动，5 表示专家级工作。
3
预计时间有经验贡献者完成调查、实现、测试并准备 pull request 的粗略时间范围。
1-3 hours
活动状态议题当前的可参与程度：新鲜、活跃、陈旧、阻塞或等待维护者输入。
stale
清晰度议题是否清楚说明期望改动、验收标准和下一步。
mostly clear
前置要求
Familiarity with NNI pruning moduleBasic PyTorch knowledge
新手友好度1-100 的估计分数，表示该议题对首次贡献者的友好程度。
30
研究方向
The issue asks about adding a new pruning method (e.g., HRank) to NNI. The user wants to know which base class to extend and where to implement it. Refer to the abstract pruner in nni/compression/pytorch/pruning/basic pruner.py and the existing pruners like NormPruner. The research should explore how to create a new pruner class that inherits from the base and registers it. Also check the latest NNI documentation and examples for custom pruners.
Does nni contain some new pruning base methods？ · microsoft/nni#4567 | Good First Issue