Skip to content

Repository files navigation

pixelCNN

Torch implementation of PixelCNN for MNIST. Implements the network described in thesepapers. 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

About

Torch implementation of PixelCNN for MNIST

Resources

Stars

29 stars

Watchers

2 watching

Forks

Releases

Packages

Contributors

Languages