diff --git a/tests/python/mkl/test_mkldnn.py b/tests/python/mkl/test_mkldnn.py index e597d0f5fc58..17fc29c81114 100644 --- a/tests/python/mkl/test_mkldnn.py +++ b/tests/python/mkl/test_mkldnn.py @@ -281,7 +281,6 @@ def check_pooling_training(stype): check_pooling_training(stype) -@unittest.skip("Flaky test: https://github.com/apache/incubator-mxnet/issues/12377") @with_seed() def test_activation(): def check_activation_training(stype): @@ -292,7 +291,7 @@ def check_activation_training(stype): in_location = [mx.nd.array(data_tmp).tostype(stype)] test = mx.symbol.Activation(data, act_type="relu") - check_numeric_gradient(test, in_location, numeric_eps=1e-2, rtol=0.16, atol=1e-4) + check_numeric_gradient(test, in_location, numeric_eps=1e-6, rtol=0.16, atol=1e-4) stypes = ['row_sparse', 'default'] for stype in stypes: