gated_pixelcnn

Gated PixelCNN

Tensorflow 2 implementation of the GatedPixelCNN.

Usage

The model can be trained on MNIST or CIFAR10

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

The model can be trained unconditionally or trained conditionally on class indexes one-hot representations. To train conditionnaly use the -c option.

python train.py -c

To see all training options use

python train.py -h