I forgot to mention that I only have one channel (while the tutorial uses images with three channels). Could this be causing some kind of problem?
The real_cpu size appears to be [20, 64, 64], which seems to not be enough dimensions. However, I only have one channel anyway. Do you know if the discriminator and generation classes can handle this difference in channel number, and if not, what changes I can make? Thanks.