prateekm08 / CycleGAN-straightforward

PyTorch implementation of CycleGAN.
14 stars 3 forks source link

Code crashing #1

Open Scienceseb opened 6 years ago

Scienceseb commented 6 years ago

Hi jus to let you know, that with a batch size of 1 the code crash everytime at epoch 11...but if you change the batch size no more crashing. It crash in the line: score += discriminator(image[:, :, xp:x, yp:y]) of the pass_through_discriminator(discriminator, image)

ghost commented 5 years ago

@Scienceseb - what batch size worked for you ??

Scienceseb commented 5 years ago

8 and 16