pytorch/examples

Changes needed to run DCGAN 32x32

Open

#486 aperta il 29 dic 2018

Vedi su GitHub
 (1 commento) (4 reazioni) (0 assegnatari)Python (9429 fork)batch import
good first issuevision

Metriche repository

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

Descrizione

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?

Guida contributor

Direzione di ricerca
Investigare l'implementazione DCGAN in pytorch/examples, in particolare i layer del generatore e del discriminatore. Verificare l'architettura corretta per input 32x32 regolando le dimensioni del kernel, stride e padding per garantire che i tensori di output corrispondano alle dimensioni previste. Tracciare il forward pass per identificare la fonte esatta degli errori di disallineamento delle dimensioni.
Tech stack
pythonpytorch
Dominio
machine learningcomputer vision
Tipo issue
Bug
DifficoltàQuanto dovrebbe essere impegnativa per un nuovo contributor.
2
Tempo stimatoTempo stimato per completare e verificare uno scope piccolo.
1-3 ore
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
PythonPyTorchGit
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
50