Modify convRBM