pytorch/examples

Changes needed to run DCGAN 32x32

Open

#486 ouverte le 29 déc. 2018

Voir sur GitHub
 (1 commentaire) (4 réactions) (0 assignés)Python (9 429 forks)batch import
good first issuevision

Métriques du dépôt

Stars
 (21 634 stars)
Métriques de merge PR
 (Aucune PR mergée en 30 j)

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?

Guide contributeur

Direction de recherche
Examinez l'implémentation DCGAN dans pytorch/examples, en particulier les couches du générateur et du discriminateur. Vérifiez l'architecture correcte pour les entrées 32x32 en ajustant les tailles de noyau, les pas et le rembourrage pour garantir que les tenseurs de sortie correspondent aux dimensions attendues. Tracez la propagation avant pour identifier la source exacte des erreurs de décalage de taille.
Stack technique
pythonpytorch
Domaine
machine learningcomputer vision
Type d'issue
Bug
DifficultéDifficulté estimée pour un nouveau contributeur, de 1 pour un très petit changement à 5 pour un travail expert.
2
Temps estiméFourchette de temps approximative pour investiguer, implémenter, tester et préparer une pull request.
1-3 heures
Statut d'activitéDisponibilité apparente de l'issue : fraîche, active, ancienne, bloquée ou en attente d'un mainteneur.
Active
ClartéClarté avec laquelle l'issue explique le changement attendu, les critères d'acceptation et la prochaine étape.
Plutôt claire
Prérequis
PythonPyTorchGit
Accessibilité débutantScore de 1 à 100 estimant l'accessibilité de cette issue pour un premier contributeur.
50