Merge pull request #105 from drnikolaev/fix2

weight updates should be done with beta=1
NVIDIA · Jan 21, 2016 · b9b83d4 · b9b83d4
2 parents 2e4d2ae + 4440513
commit b9b83d4
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,