Tensorflow 2 implementation of PixelSNAIL.

Train on MNIST/CIFAR10 with

python train.py -d mnist
python train.py -d cifar10

See available training options with

python train.py -h

Training data and visualisations can be seen with

tensorboard --logdir logs