yizhe-ang / vi-lab

Variational Inference and Normalizing Flow Experiments in PyTorch

vi-lab

Variational Autoencoder and Normalizing Flow Experiments in PyTorch, powered by PyTorch Lightning, nflows and Weights & Biases.

Installation

1. Install Python 3.7 and PyTorch 1.6.0
2. pip install -r requirements.txt

Usage

python run.py -c configs/config.yaml