microsoft/nni

Does nni contain some new pruning base methods？

Open

#4 567 ouverte le 21 févr. 2022

Voir sur GitHub
 (6 commentaires) (0 réactions) (1 assigné)Python (13 504 stars) (1 830 forks)batch import
good first issuemodel compressionnew feature

Description

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?:

Guide contributeur

Stack technique
pythonpytorch
Domaine
machine learningtooling
Type d'issue
research
DifficultéDifficulté estimée pour un nouveau contributeur, de 1 pour un très petit changement à 5 pour un travail expert.
3
Temps estiméFourchette de temps approximative pour investiguer, implémenter, tester et préparer une pull request.
1-3 hours
Statut d'activitéDisponibilité apparente de l'issue : fraîche, active, ancienne, bloquée ou en attente d'un mainteneur.
stale
ClartéClarté avec laquelle l'issue explique le changement attendu, les critères d'acceptation et la prochaine étape.
mostly clear
Prérequis
Familiarity with NNI pruning moduleBasic PyTorch knowledge
Accessibilité débutantScore de 1 à 100 estimant l'accessibilité de cette issue pour un premier contributeur.
30
Direction de recherche
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.