未验证 提交 5fc8de44 编写于 作者: Q qingqing01 提交者: GitHub

Remove warning in batch_norm_op (#19260)

上级 d08d5ab5
...@@ -245,8 +245,8 @@ class BatchNormKernel<platform::CPUDeviceContext, T> ...@@ -245,8 +245,8 @@ class BatchNormKernel<platform::CPUDeviceContext, T>
variance_out->mutable_data<T>(ctx.GetPlace()), C); variance_out->mutable_data<T>(ctx.GetPlace()), C);
if ((N * sample_size) == 1) { if ((N * sample_size) == 1) {
LOG(WARNING) << "Only 1 element in normalization dimension, " // Only 1 element in normalization dimension,
<< "we skip the batch norm calculation, let y = x."; // we skip the batch norm calculation, let y = x.
framework::TensorCopy(*x, ctx.GetPlace(), y); framework::TensorCopy(*x, ctx.GetPlace(), y);
return; return;
} }
......
...@@ -152,8 +152,8 @@ class BatchNormKernel<platform::CUDADeviceContext, T> ...@@ -152,8 +152,8 @@ class BatchNormKernel<platform::CUDADeviceContext, T>
functor(dev_ctx, saved_variance, static_cast<BatchNormParamType<T>>(0)); functor(dev_ctx, saved_variance, static_cast<BatchNormParamType<T>>(0));
if ((N * H * W * D) == 1) { if ((N * H * W * D) == 1) {
LOG(WARNING) << "Only 1 element in normalization dimension, " // Only 1 element in normalization dimension,
<< "we skip the batch norm calculation, let y = x."; // skip the batch norm calculation, let y = x.
framework::TensorCopy(*x, ctx.GetPlace(), y); framework::TensorCopy(*x, ctx.GetPlace(), y);
} else { } else {
double this_factor = 1. - momentum; double this_factor = 1. - momentum;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册