fix cudnn conv legacy bug (#96)

yjxiong · Aug 4, 2016 · c0d200b · c0d200b · antran89 · Oct 24, 2016
1 parent 42b9db8
commit c0d200b
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
@@ -66,12 +66,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();