From 5a97cf5c92dbee098850a40b9fb201482175e0c3 Mon Sep 17 00:00:00 2001 From: Chaitanya Prakash Bapat Date: Sat, 16 Nov 2019 22:55:04 -0800 Subject: [PATCH] bump up atol for gradient check (#16843) --- tests/python/unittest/test_gluon.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/python/unittest/test_gluon.py b/tests/python/unittest/test_gluon.py index 5d15b27fa7ea..3a56dc887647 100644 --- a/tests/python/unittest/test_gluon.py +++ b/tests/python/unittest/test_gluon.py @@ -1672,7 +1672,7 @@ def test(net, x): assert_almost_equal(y1.asnumpy(), y2.asnumpy(), rtol=1e-3, atol=1e-5) for key in grads1: - assert_almost_equal(grads1[key].asnumpy(), grads2[key].asnumpy(), rtol=1e-3, atol=1e-5) + assert_almost_equal(grads1[key].asnumpy(), grads2[key].asnumpy(), rtol=1e-3, atol=1e-4) @with_seed() def test_hybrid_static_memory():