Introduction

A repository that has implemented Varial Auto Encoder (VAE) and Adversarial Auto Encoder (AAE) using PyTorch. And this code contains code and examples for visualization.

Environment

• Python - 3.8.2
• Pytorch - 1.12.1
• Matplotlib - 3.7.1

Included Code

• VAE, AAE Network
• Loss function for learning VAE and AAE
• Distribution Generator (Uniform, Gaussian, Gaussian Mixture, Swiss Roll)

Result

• VAE

• AAE (Gaussian Mixture) [Trained Distribution / Prior Distribution]

TODO

• CVAE
• Save Network