Need help with wae?
Click the “chat” button below for chat support from the developer who created it, or find similar developers for support.

About the developer

tolstikhin
464 Stars 91 Forks BSD 3-Clause "New" or "Revised" License 44 Commits 6 Opened issues

Description

Wasserstein Auto-Encoders

Services available

!
?

Need anything else?

Contributors list

# 199,168
Python
44 commits

Repository info

This project implements an unsupervised generative modeling technique called Wasserstein Auto-Encoders (WAE), proposed by Tolstikhin, Bousquet, Gelly, Schoelkopf (2017).

Repository structure

wae.py - everything specific to WAE, including encoder-decoder losses, various forms of a distribution matching penalties, and training pipelines

run.py - master script to train a specific model on a selected dataset with specified hyperparameters

Example of output pictures

The following picture shows various characteristics of the WAE-MMD model trained on CelebA after 50 epochs:

WAE-MMD progress

We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.