diff --git a/paddle/fluid/operators/batch_norm_op.cu b/paddle/fluid/operators/batch_norm_op.cu index 6fc78732b1063af04a34de5d690a4f2ed75978f2..1758463141cb8f9510b7d7f8d62e69f0ce0e4013 100644 --- a/paddle/fluid/operators/batch_norm_op.cu +++ b/paddle/fluid/operators/batch_norm_op.cu @@ -382,8 +382,8 @@ class BatchNormKernel } // Run training mode. - // obtain running mean and running inv var, and see if we need to - // initialize them. + // obtain running mean and running inv var, and there is no need + // to initialize them. auto *mean_out = ctx.Output("MeanOut"); auto *variance_out = ctx.Output("VarianceOut"); @@ -394,10 +394,6 @@ class BatchNormKernel auto *saved_variance = ctx.Output("SavedVariance"); saved_mean->mutable_data>(ctx.GetPlace()); saved_variance->mutable_data>(ctx.GetPlace()); - math::SetConstant> - functor; - functor(dev_ctx, saved_mean, static_cast>(0)); - functor(dev_ctx, saved_variance, static_cast>(0)); if ((N * H * W * D) == 1) { // Only 1 element in normalization dimension,