First modification
Parallelise convolutional rbm