/pytorch-vq-vae

PyTorch implementation of VQ-VAE by Aäron van den Oord et al.

Primary LanguageJupyter NotebookMIT LicenseMIT

pytorch-vq-vae

PyTorch implementation of VQ-VAE by Aäron van den Oord et al.

Requirements

$ pip3 install -r requirements.txt

Notebook

See vq-vae.ipynb