From a40c2a08421ebf9a164e198a70752f2d5cb1c93d Mon Sep 17 00:00:00 2001 From: Jonathan L Long Date: Sun, 20 Sep 2015 14:20:28 -0700 Subject: fix broken DeconvolutionLayer GPU backward caused by typo --- src/caffe/layers/deconv_layer.cu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'src') diff --git a/src/caffe/layers/deconv_layer.cu b/src/caffe/layers/deconv_layer.cu index ea83f56f..5dbdcc31 100644 --- a/src/caffe/layers/deconv_layer.cu +++ b/src/caffe/layers/deconv_layer.cu @@ -51,7 +51,7 @@ void DeconvolutionLayer::Backward_gpu(const vector*>& top, } // gradient w.r.t. bottom data, if necessary. if (propagate_down[i]) { - this->forward_gpu_gemm(top_diff + this->top_dim_, weight, + this->forward_gpu_gemm(top_diff + n * this->top_dim_, weight, bottom_diff + n * this->bottom_dim_, this->param_propagate_down_[0]); } -- cgit v1.2.3