Adversarial-Autoencoder
Adversarial-Autoencoder copied to clipboard
neale
An adversarial autoencoder implementation in pytorch
Hi! Thanks for the very neat training code! In the generator training part (line 142 of train.py). Shouldn't you be updating the parameters of encoder network instead of the decoder...
Thank you for this implementation of AAE. Can you share the FID scores your AAE get on MNIST, CelebA and Cifar-10?
Pytorch merged a patch which was halting development of multi-GPU training code. This was keeping v1.0 from being fully functional. Should validate that we don't get a loss explosion with...
The decoder and discriminator are being treated as a GAN, which is incorrect in the implementation of an adversarial autoencoder. In your code, you train the discriminator to distinguish between...
I tried to run the code for mnist. Something goes wrong with loading the images in the function: def inf_train_gen(train_gen): while True: for i, (images, _) in enumerate(train_gen): # yield...
