diff --git a/src/caffe/layers/cudnn_batch_norm_layer.cu b/src/caffe/layers/cudnn_batch_norm_layer.cu index bacb80e9b09..cddfa7051aa 100644 --- a/src/caffe/layers/cudnn_batch_norm_layer.cu +++ b/src/caffe/layers/cudnn_batch_norm_layer.cu @@ -87,7 +87,7 @@ void CuDNNBatchNormLayer::Backward_gpu( cudnn::dataType::zero, #if CUDNN_VERSION >= 4005 cudnn::dataType::one, - cudnn::dataType::zero, + cudnn::dataType::one, #endif bottom_desc_, bottom_data,