summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJeff Donahue <jeff.donahue@gmail.com>2015-09-20 22:16:59 (GMT)
committerJeff Donahue <jeff.donahue@gmail.com>2015-09-20 22:16:59 (GMT)
commit9dca961545caf3c0d4321fc580d352f5052af6fb (patch)
tree30b8d949853cf5b9f227a6a33d4d008070a4af6c
parent68655b55925cfa5cd6543cdfb879bf6e68bd2a3c (diff)
parenta40c2a08421ebf9a164e198a70752f2d5cb1c93d (diff)
downloadcaffeonacl-9dca961545caf3c0d4321fc580d352f5052af6fb.zip
caffeonacl-9dca961545caf3c0d4321fc580d352f5052af6fb.tar.gz
caffeonacl-9dca961545caf3c0d4321fc580d352f5052af6fb.tar.bz2
Merge pull request #3095 from longjon/fix-deconv-backward
Fix broken DeconvolutionLayer GPU backward caused by ND conv 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]);
}