DiscoGAN-pytorch
DiscoGAN-pytorch copied to clipboard
Width and Height of images needs to be identical
Looks like you crash if width and height isn't equal on custom datasets. I worked around by forcing my 640x360 images to 640x640.
Once I set '-- input_scale_size' not to 64, an error will be reported. It seems that the discriminator cannot adapt to the size of the input image. How did you solve the problem? "ValueError: Target and input must have the same number of elements. target nelement (2) != input nelement (338)"