size mismatch for block4.c2.weight

[sumorday]

Dear Kwotsin:

Hi, When I use the dataset

LSUN-Bedroom | Train | 128 x 128

The data is 128x128 size, right?

So I changed the example file in https://github.com/kwotsin/mimicry/blob/master/examples/sngan_example.py

the original one it below:

# Define models and optimizers
netG = sngan.SNGANGenerator32().to(device)
netD = sngan.SNGANDiscriminator32().to(device)

I guess I have to change it into 128, that is:

netG = sngan.SNGANGenerator128().to(device)
netD = sngan.SNGANDiscriminator128().to(device)

But when I changed it to 128 , the error happened(size mismatch)
If I keep 32, the original one. the code can be running without any issues...
Did I missed anything? Looking forward to getting your feedback. Thank you in advance!!!

Best regards,
Edward