diff options
| -rw-r--r-- | src/caffe/layers/deconv_layer.cu | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/caffe/layers/deconv_layer.cu b/src/caffe/layers/deconv_layer.cu index 39bc4de8..8a1eed8a 100644 --- a/src/caffe/layers/deconv_layer.cu +++ b/src/caffe/layers/deconv_layer.cu @@ -52,7 +52,8 @@ void DeconvolutionLayer<Dtype>::Backward_gpu(const vector<Blob<Dtype>*>& top, // gradient w.r.t. bottom data, if necessary. if (propagate_down[i]) { this->forward_gpu_gemm(top_diff + top[i]->offset(n), weight, - bottom_diff + bottom[i]->offset(n)); + bottom_diff + bottom[i]->offset(n), + this->param_propagate_down_[0]); } } } |