Vector Quantized VAEs

A Pytorch implementation of this paper

This repository builds off this one

Running the following command gets ~ 5.05 BPD ( Instead of 4.7 / 4.8, as is reported in said papers 1 2

python main.py --hH 16 --n_codebooks 2 --embed_dim 256 --n_res_channels 256 --n_channels 256 --batch_size 256 --lr 5e-4

Contribute

All contributions / comments / remarks are highly welcomed.