Skip to content

Latest commit

 

History

History
17 lines (10 loc) · 1.17 KB

README.md

File metadata and controls

17 lines (10 loc) · 1.17 KB
Raw

Introduction

This repo is the Wasserstein Auto-Encoders for Text and Image Generation from this paper in Pytorch.

Feature

  • Transformer based (from scratch / from Huggingface) Wasserstein Auto-Encoder with MMD metric (WAE-MMD) for text generation
  • Transformer based (from scratch / from Huggingface) Wasserstein Auto-Encoder with adversarial metric (WAE-adv) for text generation
  • Transformer based (from scratch / from Huggingface) Wasserstein Auto-Encoder with MMD metric and gradient penalty (WAE-GP-MMD) for text generation
  • Transformer based (from scratch / from Huggingface) Wasserstein Auto-Encoder with adversarial metric and gradient penalty (WAE-GP-adv) for text generation
  • Wasserstein Auto-Encoder with adversarial metric (WAE-adv) for image generation (based on https://github.com/schelotto/Wasserstein-AutoEncoders)
  • Wasserstein Auto-Encoder with MMD metric (WAE-MMD) for image generation (based on https://github.com/schelotto/Wasserstein-AutoEncoders)

in progress...