microsoft/nni

Does nni contain some new pruning base methods？

Open

#4567 aperta il 21 feb 2022

Vedi su GitHub
 (6 commenti) (0 reazioni) (1 assegnatario)Python (1830 fork)batch import
good first issuemodel compressionnew feature

Metriche repository

Star
 (13.504 star)
Metriche merge PR
 (Nessuna PR mergiata in 30 g)

Descrizione

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

Guida contributor

Direzione di ricerca
Il problema riguarda l'aggiunta di un nuovo metodo di pruning (ad esempio HRank) a NNI. L'utente vuole sapere quale classe base estendere e dove implementarlo. Fare riferimento al pruner astratto in nni/compression/pytorch/pruning/basic pruner.py e ai pruner esistenti come NormPruner. La ricerca dovrebbe esplorare come creare una nuova classe pruner che erediti dalla classe base e la registri. Controllare anche la documentazione più recente di NNI e gli esempi per i pruner personalizzati.
Tech stack
pythonpytorch
Dominio
machine learningai
Tipo issue
Funzionalità
DifficoltàQuanto dovrebbe essere impegnativa per un nuovo contributor.
3
Tempo stimatoTempo stimato per completare e verificare uno scope piccolo.
1-2 giorni
Stato attivitàQuanto è probabile che la issue sia ancora attiva e reviewable.
Attiva
ChiarezzaQuanto chiaramente la issue descrive problema, scope e risultato atteso.
Abbastanza chiara
Prerequisiti
PythonPyTorchModel Pruning Concepts
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
60