projects
/
platform
/
upstream
/
caffe.git
/ commitdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
| commitdiff |
tree
raw
|
patch
| inline |
side by side
(parent:
aa2a6f5
)
DeconvolutionLayer Backward_gpu fix: don't redo im2col
author
Jeff Donahue
<jeff.donahue@gmail.com>
Thu, 20 Aug 2015 18:54:08 +0000
(11:54 -0700)
committer
Jeff Donahue
<jeff.donahue@gmail.com>
Thu, 20 Aug 2015 18:58:41 +0000
(11:58 -0700)
src/caffe/layers/deconv_layer.cu
patch
|
blob
|
history
diff --git
a/src/caffe/layers/deconv_layer.cu
b/src/caffe/layers/deconv_layer.cu
index
39bc4de
..
8a1eed8
100644
(file)
--- 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]);
}
}
}