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
All contributions / comments / remarks are highly welcomed.