pytorch/examples

Changes needed to run DCGAN 32x32

Open

#486 opened on 2018年12月29日

GitHub で見る
 (1 comment) (4 reactions) (0 assignees)Python (9,429 forks)batch import
good first issuevision

Repository metrics

Stars
 (21,634 stars)
PR merge metrics
 (30d に merged PR はありません)

説明

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?

コントリビューターガイド