This repo is an implementation of VQ-VAE using PyTorch This implementation is not official, this repo is just for study!
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.
- 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.