accumulate gradients in cudnn conv layer

BVLC · Feb 26, 2015 · 836a2d9 · 836a2d9
1 parent f249bed
commit 836a2d9
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/src/caffe/layers/cudnn_conv_layer.cu b/src/caffe/layers/cudnn_conv_layer.cu
@@ -54,12 +54,10 @@ void CuDNNConvolutionLayer<Dtype>::Backward_gpu(const vector<Blob<Dtype>*>& top,
  if (this->param_propagate_down_[0]) {
  weight = this->blobs_[0]->gpu_data();
  weight_diff = this->blobs_[0]->mutable_gpu_diff();
- caffe_gpu_set(this->blobs_[0]->count(), Dtype(0), weight_diff);
  }
  Dtype* bias_diff = NULL;
  if (this->bias_term_ && this->param_propagate_down_[1]) {
  bias_diff = this->blobs_[1]->mutable_gpu_diff();
- caffe_gpu_set(this->blobs_[1]->count(), Dtype(0), bias_diff);
  }
  for (int i = 0; i < top.size(); ++i) {
  const Dtype* top_diff = top[i]->gpu_diff();