PS-- pylearn2/scripts/train_example uses this class, so changing it can change the optimization. Can you check that it still works OK after this change? If it runs for a really long time now we might need to change the config so that the buildbot doesn't get stuck.