summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJonathan L Long <jonlong@cs.berkeley.edu>2015-09-20 21:20:28 (GMT)
committerJonathan L Long <jonlong@cs.berkeley.edu>2015-09-20 21:20:28 (GMT)
commita40c2a08421ebf9a164e198a70752f2d5cb1c93d (patch)
tree30b8d949853cf5b9f227a6a33d4d008070a4af6c
parent68655b55925cfa5cd6543cdfb879bf6e68bd2a3c (diff)
downloadcaffeonacl-a40c2a08421ebf9a164e198a70752f2d5cb1c93d.zip
caffeonacl-a40c2a08421ebf9a164e198a70752f2d5cb1c93d.tar.gz
caffeonacl-a40c2a08421ebf9a164e198a70752f2d5cb1c93d.tar.bz2
fix broken DeconvolutionLayer GPU backward caused by typo
-rw-r--r--src/caffe/layers/deconv_layer.cu2
1 files changed, 1 insertions, 1 deletions
diff --git a/src/caffe/layers/deconv_layer.cu b/src/caffe/layers/deconv_layer.cu
index ea83f56..5dbdcc3 100644
--- a/src/caffe/layers/deconv_layer.cu
+++ b/src/caffe/layers/deconv_layer.cu
@@ -51,7 +51,7 @@ 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 + 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]);
}