DCGAN example doesn't work with different image sizes #70
Description
I'm trying to use this code as a starting point for building GANs from my own image data-- 512x512 grayscale images. If I change any of the default arguments (e.g. --imageSize 512) I get the following error:
Traceback (most recent call last): File "main.py", line 209, in <module> errD_real = criterion(output, label) File "/opt/python/lib/python3.6/site-packages/torch/nn/modules/module.py", line 210, in __call__ result = self.forward(*input, **kwargs) File "/opt/python/lib/python3.6/site-packages/torch/nn/modules/loss.py", line 36, in forward return backend_fn(self.size_average, weight=self.weight)(input, target) File "/opt/python/lib/python3.6/site-packages/torch/nn/_functions/thnn/loss.py", line 22, in forward assert input.nelement() == target.nelement() AssertionError Still learning my way around PyTorch so the network architectures that are spit out before the above message don't yet give me much intuition. I appreciate any pointers you can give!