microsoft/nni

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

Open

#4.301 aberto em 10 de nov. de 2021

Ver no GitHub
 (4 comments) (0 reactions) (1 assignee)Python (1.830 forks)batch import
enhancementexternal contributorgood first issue

Métricas do repositório

Stars
 (13.504 stars)
Métricas de merge de PR
 (Nenhuma PRs mesclada em 30d)

Description

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!

Guia do colaborador

Direção de pesquisa
Investigue como o NNI atualmente lida com a substituição de módulos para operações semelhantes (por exemplo, Conv2d, BatchNorm). Examine a implementação no módulo de compressão de modelo do NNI, provavelmente no diretório `nni/compression/pytorch/`. O erro indica que `PixelShuffle` não está na lista de módulos suportados. Você precisará adicionar suporte registrando `PixelShuffle` como um módulo substituível, de forma semelhante a outros módulos. Verifique os casos de teste existentes para substituição de módulos para garantir consistência.
Pilha de tecnologia
pythonpytorch
Domain
machine learningai
Tipo Issue
Bug
DifficultyDificuldade de implementação estimada para um novo contribuidor, de 1 para alterações muito pequenas a 5 para trabalho de nível especializado.
3
Tempo estimadoUm intervalo de tempo aproximado para um colaborador experiente investigar, implementar, testar e preparar um pull request.
Meio dia
Status da atividadeQuão disponível o issue aparece agora: novo, ativo, obsoleto, bloqueado ou aguardando entrada do mantenedor.
Ativo
ClarityCom que clareza o issue explica a mudança esperada, os critérios de aceitação e a próxima etapa.
Claro
Prerequisites
PythonPyTorch
Simpatia para novatosUma pontuação de 1 a 100 que estima o quão acessível este issue é para colaboradores iniciantes.
40