From 78d178a9ecab552751e5d9f2adc18ae33658eb4c Mon Sep 17 00:00:00 2001 From: Junyuan Xie Date: Mon, 14 May 2018 17:32:41 -0700 Subject: [PATCH] fix --- 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 96623f0c2067..32fc578c0c0b 100644 --- a/tests/python/unittest/test_gluon.py +++ b/tests/python/unittest/test_gluon.py @@ -1023,7 +1023,7 @@ def test(net, x): y1, grads1 = test(net1, x) y2, grads2 = test(net2, x) - assert_array_equal(y1.asnumpy(), y2.asnumpy()) + 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)