Intro

This GitHub repository features a PyTorch implementation of VAE and Pi-VAE (paper link) for archival and evaluation purpose of the final project of Math 4210 Mathematical Foundations of Machine Learning, taken in Fall 2023, at Georgia Tech.

Installation

# clone this repo
source make_env_ivae.sh

Train and Eval on the Swiss Roll Dataset

source init_env_ivae.sh
cd model_training_directory
source run_swiss_roll.sh

Train and Eval on the Swiss Roll Dataset

source init_env_ivae.sh
cd model_training_directory
source run_rat_hippo.sh