DeconvolutionLayer Backward_gpu fix: don't redo im2col
authorJeff Donahue <jeff.donahue@gmail.com>
Thu, 20 Aug 2015 18:54:08 +0000 (11:54 -0700)
committerJeff Donahue <jeff.donahue@gmail.com>
Thu, 20 Aug 2015 18:58:41 +0000 (11:58 -0700)
src/caffe/layers/deconv_layer.cu

index 39bc4de..8a1eed8 100644 (file)
@@ -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]);
         }
       }
     }