/vq-vae-pytorch

VQ-VAE implementation pytorch

Primary LanguagePython

VQ-VAE

This is a lightweight (200 loc) implementation of the VQ-VAE Neural Discrete representation learning.

Training

By default it trains on cifar10

python vq-vae-img.py

Edit hyperparams, paths in source code to train on ImageNet

ImageNet reconstruction after 40k iterations (K=512, D=128)

ImageNet

License

MIT