/VectorQuantizedVAE

A PyTorch implementation of "Continuous Relaxation Training of Discrete Latent Variable Image Models"

Primary LanguageJupyter NotebookMIT LicenseMIT

Watchers