Performance on MNIST

[koriavinash1]

@lucidrains Thanks for this repository.

I'm trying this script for generating Morpho-MNIST digits (MNIST with tweaking morphology like thickness and intensity), image size 32x32, 3 channels.

Even after training for approximately upto 200k I'm unable to get FID < 60. I'm attaching some of the generated images, it would be really helpful if you could provide some insights on this.

Thank you, Avinash