pytorch/examples

Changes needed to run DCGAN 32x32

Open

#486 创建于 2018年12月29日

在 GitHub 查看
 (1 评论) (4 反应) (0 负责人)Python (9,429 fork)batch import
good first issuevision

仓库指标

Star
 (21,634 star)
PR 合并指标
 (30 天内没有已合并 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?

贡献者指南