schelotto / Wasserstein Autoencoders
Licence: mit
PyTorch implementation of Wasserstein Auto-Encoders
Stars: ✭ 230
Programming Languages
python
139335 projects - #7 most used programming language
Labels
Introduction
This is the implementation of Wasserstein Auto-Encoders paper in PyTorch.
I rewrite the model in DC-GAN for the encoder/decoder/discriminator instead of simple MLP
Requirement
- python 3
- PyTorch >= 0.4
- torchvision
- numpy
- tqdm
Train
- To train a WAE-GAN:
python wae_gan.py
- To train a WAE-MMD:
python wae_mmd.py
Randomly Generated Images (100 Epochs)
WAE-MMD
WAE-GAN
Note that the project description data, including the texts, logos, images, and/or trademarks,
for each open source project belongs to its rightful owner.
If you wish to add or remove any projects, please contact us at [email protected].