microsoft/nni

RuntimeError: Has not supported replacing the module: `PixelShuffle`

Open

#4301 aperta il 10 nov 2021

Vedi su GitHub
 (4 commenti) (0 reazioni) (1 assegnatario)Python (1830 fork)batch import
enhancementexternal contributorgood first issue

Metriche repository

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

Descrizione

Describe the issue:

Environment:

  • NNI version:Release 2.3 (github master branch)
  • Training service (local|remote|pai|aml|etc):local
  • Client OS:Ubuntu 18.04
  • Server OS (for remote mode only):
  • Python version: Python 3.7.0
  • PyTorch/TensorFlow version: 1.7.1
  • Is conda/virtualenv/venv used?: virutalenv
  • Is running in Docker?: no

For super resolution models, nn.PixelShuffle(scale) is widely used. When I tried to compress and speed up a super resolution model with nn.PixelShuffle( ), an error was met: RuntimeError: Has not supported replacing the module: PixelShuffle . Is there a way that I can fix or work around it. Thanks very much!

Guida contributor

Direzione di ricerca
Indaga come NNI gestisce attualmente la sostituzione dei moduli per operazioni simili (ad esempio, Conv2d, BatchNorm). Esamina l'implementazione nel modulo di compressione del modello di NNI, probabilmente nella directory `nni/compression/pytorch/`. L'errore indica che `PixelShuffle` non è nell'elenco dei moduli supportati. Sarà necessario aggiungere il supporto registrando `PixelShuffle` come modulo sostituibile, in modo simile ad altri moduli. Controlla i casi di test esistenti per la sostituzione dei moduli per garantire coerenza.
Tech stack
pythonpytorch
Dominio
machine learningai
Tipo issue
Bug
DifficoltàQuanto dovrebbe essere impegnativa per un nuovo contributor.
3
Tempo stimatoTempo stimato per completare e verificare uno scope piccolo.
Mezza giornata
Stato attivitàQuanto è probabile che la issue sia ancora attiva e reviewable.
Attiva
ChiarezzaQuanto chiaramente la issue descrive problema, scope e risultato atteso.
Chiara
Prerequisiti
PythonPyTorch
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
40