/vqvae-pytorch-impl

Primary LanguageJupyter Notebook

VQ-VAE Implementation in PyTorch

This repo is an implementation of VQ-VAE using PyTorch This implementation is not official, this repo is just for study!

Experiments

All code about training and inference is in "train_vqvae-*.ipynb" files. I trained VQ-VAE with STL-10 dataset.

Following pictures are result of experiments. All pictures are test image.

  • VQ loss version
    Image
  • EMA version
    Image

Reference

  • Oord, A., Oriol Vinyals and K. Kavukcuoglu. “Neural Discrete Representation Learning.” NIPS (2017).
  • Adam Coates, Honglak Lee, Andrew Y. Ng An Analysis of Single Layer Networks in Unsupervised Feature Learning AISTATS, 2011.