Skip to content

Latest commit

 

History

History
17 lines (15 loc) · 550 Bytes

README.md

File metadata and controls

17 lines (15 loc) · 550 Bytes

pixelCNN

Torch implementation of PixelCNN for MNIST. Implements the network described in these papers. kundan2510's Theano implementation was an invaluable reference.

Requires cutorch and cudnn.

Also requires the MNIST dataset (duh!):

luarocks install mnist

Train the model (this will automatically save the model every so often):

th trainMNIST.lua

Generate images using the saved model:

th genMNIST.lua