fix atol for test_preloaded_multi_sgd (#16356)

apache · Oct 2, 2019 · 8136d49 · 8136d49
1 parent 480b50c
commit 8136d49
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/python/gpu/test_operator_gpu.py b/tests/python/gpu/test_operator_gpu.py
@@ -396,7 +396,7 @@ def _assert_all_almost_equal(lhs_list, rhs_list, rtol, atol):
             assert_almost_equal(lhs.asnumpy(), rhs.asnumpy(), rtol=rtol, atol=atol)
     if dtype == 'float16':
         rtol = 1e-3
-        atol = 1e-3
+        atol = 1e-2
     else:
         rtol = 1e-5
         atol = 1e-6