A few extensions to cuda-convnet