diff options
| author | Jonathan L Long <jonlong@cs.berkeley.edu> | 2015-09-20 14:20:28 -0700 |
|---|---|---|
| committer | Jonathan L Long <jonlong@cs.berkeley.edu> | 2015-09-20 14:20:28 -0700 |
| commit | a40c2a08421ebf9a164e198a70752f2d5cb1c93d (patch) | |
| tree | 30b8d949853cf5b9f227a6a33d4d008070a4af6c /src | |
| parent | 68655b55925cfa5cd6543cdfb879bf6e68bd2a3c (diff) | |
| download | caffeonacl-a40c2a08421ebf9a164e198a70752f2d5cb1c93d.tar.gz caffeonacl-a40c2a08421ebf9a164e198a70752f2d5cb1c93d.tar.bz2 caffeonacl-a40c2a08421ebf9a164e198a70752f2d5cb1c93d.zip | |
fix broken DeconvolutionLayer GPU backward caused by typo
Diffstat (limited to 'src')
| -rw-r--r-- | src/caffe/layers/deconv_layer.cu | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/caffe/layers/deconv_layer.cu b/src/caffe/layers/deconv_layer.cu index ea83f56f..5dbdcc31 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]); } |