Skip to content

BenJamesbabala/pixelCNN-1

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

About

Torch implementation of PixelCNN for MNIST

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Lua 100.0%