From fcfb0afeaec797e6d0a8214f9e1ed22a1d806124 Mon Sep 17 00:00:00 2001 From: Guoxia Wang Date: Fri, 17 Sep 2021 19:17:08 +0800 Subject: [PATCH] fix unittest (#35808) --- python/paddle/fluid/layers/nn.py | 2 +- python/paddle/fluid/tests/unittests/test_norm_op.py | 7 +++++-- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/python/paddle/fluid/layers/nn.py b/python/paddle/fluid/layers/nn.py index 13a1c8e079..06a14295a8 100755 --- a/python/paddle/fluid/layers/nn.py +++ b/python/paddle/fluid/layers/nn.py @@ -5061,7 +5061,7 @@ def l2_normalize(x, axis, epsilon=1e-12, name=None): X = paddle.randn(shape=[3, 5], dtype='float64') out = paddle.fluid.layers.l2_normalize(X, axis=-1) - print(out.numpy()) + print(out) # [[ 0.21558504 0.56360189 0.47466096 0.46269539 -0.44326736] # [-0.70602414 -0.52745777 0.37771788 -0.2804768 -0.04449922] diff --git a/python/paddle/fluid/tests/unittests/test_norm_op.py b/python/paddle/fluid/tests/unittests/test_norm_op.py index af81d15717..6b424e03cc 100644 --- a/python/paddle/fluid/tests/unittests/test_norm_op.py +++ b/python/paddle/fluid/tests/unittests/test_norm_op.py @@ -41,10 +41,10 @@ class TestNormOp(OpTest): self.outputs = {'Out': y, 'Norm': norm} def test_check_output(self): - self.check_output(atol=1e-5) + self.check_output() def test_check_grad(self): - self.check_grad(['X'], 'Out', max_relative_error=0.008) + self.check_grad(['X'], 'Out') def init_test_case(self): self.shape = [2, 3, 4, 5] @@ -97,6 +97,9 @@ class TestNormOp6(TestNormOp): def init_dtype(self): self.dtype = "float32" + def test_check_grad(self): + self.check_grad(['X'], 'Out', max_relative_error=0.008) + @unittest.skipIf(not fluid.core.is_compiled_with_cuda(), "core is not compiled with CUDA") -- GitLab