diff options
| author | Jeff Donahue <jeff.donahue@gmail.com> | 2015-08-20 11:54:08 -0700 |
|---|---|---|
| committer | Jeff Donahue <jeff.donahue@gmail.com> | 2015-08-20 11:58:41 -0700 |
| commit | 35657e31ad82f73a0682deb52cb1606b33a202be (patch) | |
| tree | f45b261480156940882cc8c71891080fd48b65eb | |
| parent | aa2a6f55b9e50b29d607aaee0fae19bd085d6565 (diff) | |
| download | caffeonacl-35657e31ad82f73a0682deb52cb1606b33a202be.tar.gz caffeonacl-35657e31ad82f73a0682deb52cb1606b33a202be.tar.bz2 caffeonacl-35657e31ad82f73a0682deb52cb1606b33a202be.zip | |
DeconvolutionLayer Backward_gpu fix: don't redo im2col
| -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]); } } } |