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?