/MinVQVAE

Minimal Discrete Variational Autoencoder (VQ-VAE) implementation in PyTorch, train, eval and test on Cifar10, CelebA, and ImageNet ILSVRC2012, get good result.

Primary LanguagePythonGNU General Public License v3.0GPL-3.0

Stargazers