VQ-VAE - Pytorch Implementation of Neural Discrete Representation Learning by Aaron van den Oord , Oriol Vinyals ,Koray Kavukcuoglu

Link :

https://arxiv.org/pdf/1711.00937.pdf

Requisite :

Python >= 3.6 PyTorch >= 1.1 Numpy

Usage :

Import the Model class from the VAE.py module into your main function . A sample main function will be put up shortly