Small optimization

Signed-off-by: Axel Davy <axel.davy@ens.fr>
Merge branch 'cluster_local' of https://github.com/axeldavy/pylearn2 into cluster_local