Modified clustergan.py to make it compatible with pytorch2.0+ and removing runtime errors of inplace operations. Tuned some of the parameters for training the clustergan model, the output folder will contain losses_epoch.png to visualise training of the Discriminator, Encoder and Generator.
baicaiPCX
commented
2 months ago
Modified clustergan.py to make it compatible with pytorch2.0+ and removing runtime errors of inplace operations. Tuned some of the parameters for training the clustergan model, the output folder will contain losses_epoch.png to visualise training of the Discriminator, Encoder and Generator.