fix

apache · May 15, 2018 · 78d178a · 78d178a
1 parent 6f2fd32
commit 78d178a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)