diff --git a/python/paddle/fluid/tests/unittests/test_group_norm_op_v2.py b/python/paddle/fluid/tests/unittests/test_group_norm_op_v2.py
index 833eeb33641c90206b4f85a9bf49ac57f5c8c8b1..47761eb5eaf8789f7575ed850d37613efdd5cfc3 100644
--- a/python/paddle/fluid/tests/unittests/test_group_norm_op_v2.py
+++ b/python/paddle/fluid/tests/unittests/test_group_norm_op_v2.py
@@ -31,24 +31,24 @@ class TestDygraphGroupNormv2(unittest.TestCase):
         if core.is_compiled_with_cuda() and core.op_support_gpu("group_norm"):
             places.append(fluid.CUDAPlace(0))
         for p in places:
-            shape = [2, 6, 2, 2]
+            shape = [2, 2, 2, 2]
 
             def compute_v1(x):
                 with fluid.dygraph.guard(p):
-                    gn = fluid.dygraph.GroupNorm(channels=6, groups=2)
+                    gn = fluid.dygraph.GroupNorm(channels=2, groups=2)
                     y = gn(fluid.dygraph.to_variable(x))
                 return y.numpy()
 
             def compute_v2(x):
                 with fluid.dygraph.guard(p):
-                    gn = paddle.nn.GroupNorm(num_channels=6, num_groups=2)
+                    gn = paddle.nn.GroupNorm(num_channels=2, num_groups=2)
                     y = gn(fluid.dygraph.to_variable(x))
                 return y.numpy()
 
             def test_weight_bias_false():
                 with fluid.dygraph.guard(p):
                     gn = paddle.nn.GroupNorm(
-                        num_channels=6,
+                        num_channels=2,
                         num_groups=2,
                         weight_attr=False,
                         bias_attr=False)
@@ -56,7 +56,7 @@ class TestDygraphGroupNormv2(unittest.TestCase):
             x = np.random.randn(*shape).astype("float32")
             y1 = compute_v1(x)
             y2 = compute_v2(x)
-            result = np.allclose(y1, y2)
+            result = np.allclose(y1, y2, atol=1e-5)
             if not result:
                 print("y1:", y1, "\ty2:", y2)
             self.assertTrue(result)