Trained VQVAE for 20 epochs + pixelSNAIL prior for 50 epochs, which took ~30m on RTX3080 laptop GPU. See config files for training details.
from top: epoch 0 -> bottom: epoch 50
given a number between 0~9, generates MNIST-like image of that digit. PixelSNAIL trains much faster with conditioning.