/PyTorch-VAE

A Collection of Variational Autoencoders (VAE) in PyTorch.

Primary LanguagePythonApache License 2.0Apache-2.0

Pytorch VAE

Work in Progress

A stripped down fork of https://github.com/AntixK/PyTorch-VAE. For a more complete package allong with results of the implementation check out the original repo.

Description

A collection of Variational AutoEncoders (VAEs) implemented in pytorch with focus on reproducibility. The aim of this project is to provide a quick and simple working example for many of the cool VAE models out there.

Requirements

  • Python >= 3.6
  • PyTorch >= 1.5.0
  • Pytorch Lightning >= 1.0.0

Installation

git clone https://github.com/threewisemonkeys-as/PyTorch-VAE.git
cd PyTorch-VAE
pip install -r requirements.txt