Pylearn2 wrapper for mnist variations