ClusterGAN error starting training

eriklindernoren / PyTorch-GAN

PyTorch implementations of Generative Adversarial Networks.

MIT License

16.41k stars 4.07k forks source link

ClusterGAN error starting training #96

Open jbmaxwell opened 4 years ago

jbmaxwell commented 4 years ago

I'm getting the error: ValueError: Target and input must have the same number of elements. target nelement (64) != input nelement (32), traceable back to line 470 in clustergan.py. I confess I'm not totally sure how to fix the problem (seems like maybe it's something caused by a PyTorch update?).

jbmaxwell commented 4 years ago

ah... wait... batch_size=32 seems to get it running. What would be the solution to allowing a larger batch size?