From the train.py, I have checked that you are not using the EMA as it is important for GAN. Have you used it when you obtained the results? Please let me know. Regards