Per-batch monitoring for Pylearn2 wrapping