pytorch/examples

Changes needed to run DCGAN 32x32

Open

#486 aberto em 29 de dez. de 2018

Ver no GitHub
 (1 comment) (4 reactions) (0 assignees)Python (9.429 forks)batch import
good first issuevision

Métricas do repositório

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

Description

I'm trying to run the DCGAN on Imagenet 32x32, but am running into problems.

If I just change the --imageSize to 32, then the convolutional layers break and I get the error RuntimeError: sizes must be non-negative. I changed the kernel size of the final Generator layer to 1 and the kernel size of the final Discriminator layer to 2 (as per @rajaswa in this related isssue) but then I get a size mismatch error ValueError: Target and input must have the same number of elements. target nelement (64) != input nelement (256). I haven't made any other changes to main.py as I want to establish a baseline model.

What other changes to the parameters/Generator/Discriminator do I need?

Guia do colaborador

Direção de pesquisa
Investigue a implementação do DCGAN em pytorch/examples, especificamente as camadas do gerador e do discriminador. Verifique a arquitetura correta para entradas 32x32 ajustando tamanhos de kernel, strides e padding para garantir que os tensores de saída correspondam às dimensões esperadas. Rastreie a passagem direta para identificar a fonte exata dos erros de incompatibilidade de tamanho.
Pilha de tecnologia
pythonpytorch
Domain
machine learningcomputer vision
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.
2
Tempo estimadoUm intervalo de tempo aproximado para um colaborador experiente investigar, implementar, testar e preparar um pull request.
1-3 horas
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.
Principalmente claro
Prerequisites
PythonPyTorchGit
Simpatia para novatosUma pontuação de 1 a 100 que estima o quão acessível este issue é para colaboradores iniciantes.
50