Skip to content

rocke2020/Wasserstein-AutoEncoders

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

34 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Introduction

This is the implementation of Wasserstein Auto-Encoders paper in PyTorch. simple and clean WAE

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_MMD

WAE-GAN

WAE_GAN

About

PyTorch implementation of Wasserstein Auto-Encoders

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 99.8%
  • Shell 0.2%