Saving is now working. Gave up on pylearn2 speedups.