diff --git a/lite/kernels/arm/group_norm_compute.cc b/lite/kernels/arm/group_norm_compute.cc index fd53f66e086cb34bfab2538c3b54c2fc8de8c32a..2e370414f4079f8dbbc2e5cc9af294c7b3f88718 100644 --- a/lite/kernels/arm/group_norm_compute.cc +++ b/lite/kernels/arm/group_norm_compute.cc @@ -44,7 +44,6 @@ void GroupNormCompute::Run() { int ngroup = n * groups; int cnt = spatial_size >> 4; int remain = spatial_size % 16; - LOG(INFO) << "param.y dims: " << param.out->dims(); // compute saved_mean and saved_variance #pragma omp parallel for for (int n = 0; n < ngroup; ++n) { diff --git a/lite/tests/kernels/group_norm_compute_test.cc b/lite/tests/kernels/group_norm_compute_test.cc index b77689a5319e06e54510db9a295e8c120e8bf645..92cdc38c9a6ca88495c403735d894ff44eff7674 100644 --- a/lite/tests/kernels/group_norm_compute_test.cc +++ b/lite/tests/kernels/group_norm_compute_test.cc @@ -75,7 +75,6 @@ class GroupNormComputeTest : public arena::TestCase { int ch_per_group = channels_ / groups_; CHECK_EQ(x->dims()[1], channels_); int spatial_size = ch_per_group * x->dims()[2] * x->dims()[3]; - LOG(INFO) << "base dims: " << y->dims(); // compute mean for (int i = 0; i < n * groups_; ++i) { const float* x_ptr = x_data + i * spatial_size; @@ -94,7 +93,6 @@ class GroupNormComputeTest : public arena::TestCase { (x_ptr[j] - saved_mean_data[i]) * (x_ptr[j] - saved_mean_data[i]); } saved_variance_data[i] = 1.f / sqrtf(sum / spatial_size + epsilon_); - LOG(INFO) << "i: " << i << ", means: " << saved_mean_data[i] << ", saved_variance_data: " << saved_variance_data[i]; } int in_size = x->dims()[2] * x->dims()[3]; // compute out @@ -151,17 +149,15 @@ void TestGroupNorm(Place place, float abs_error = 6e-5, std::vector ignored_outs = {}) { for (auto& n : {1, 3, 16}) { - for (auto& c : {1, 4, 16}) { + for (auto& c : {1}) { for (auto& h : {1, 16, 33, 56}) { - for (auto& w : {1, 17, 34, 55}) { + for (auto& w : {1, 17, 55}) { for (auto& groups: {1, 2, 4}) { if (c % groups != 0) { continue; } DDim dim_in({n, c, h, w}); float epsilon = 1e-5f; - LOG(INFO) << "input shape: " << n << ", " << c << ", " << h <<", " << w; - LOG(INFO) << "groups: " << groups; std::unique_ptr tester( new GroupNormComputeTest(place, "def", dim_in, epsilon, groups, c)); #ifdef LITE_WITH_ARM