From ecc8466365171a973b9dd30c626bbd59d69557c4 Mon Sep 17 00:00:00 2001 From: reminisce Date: Fri, 23 Mar 2018 13:14:20 -0700 Subject: [PATCH] Fix flaky unit test layer_norm --- tests/python/unittest/test_operator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/python/unittest/test_operator.py b/tests/python/unittest/test_operator.py index 240c06a5d7a2..2486be04a52f 100644 --- a/tests/python/unittest/test_operator.py +++ b/tests/python/unittest/test_operator.py @@ -2539,7 +2539,7 @@ def npy_layer_norm(data, gamma, beta, axis=1, eps=1E-5): for req in ['write', 'add']: check_numeric_gradient(out_s, {'data': data, 'gamma': gamma, 'beta': beta}, grad_nodes={'data': req, 'gamma': req, 'beta': req}, - numeric_eps=1e-2, rtol=1e-2, atol=1e-3) + numeric_eps=1e-2, rtol=1e-2, atol=1e-2) def test_layer_norm(): for dtype, forward_check_eps in zip([np.float16, np.float32, np.float64],