weight updates should be done with beta=1

NVIDIA · Jan 21, 2016 · 4440513 · 4440513
1 parent 2e4d2ae
commit 4440513
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/caffe/layers/cudnn_batch_norm_layer.cu b/src/caffe/layers/cudnn_batch_norm_layer.cu
@@ -87,7 +87,7 @@ void CuDNNBatchNormLayer<Dtype>::Backward_gpu(
       cudnn::dataType<Dtype>::zero,
 #if CUDNN_VERSION >= 4005
       cudnn::dataType<Dtype>::one,
-      cudnn::dataType<Dtype>::zero,
+      cudnn::dataType<Dtype>::one,
 #endif
       bottom_desc_,
       bottom_data,