Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
Revert "Fix flaky test: test_mkldnn.test_activation #12377 (#12418)" (#…
Browse files Browse the repository at this point in the history
…12516)

This reverts commit 445967e.
  • Loading branch information
lebeg authored and marcoabreu committed Sep 12, 2018
1 parent 90599e1 commit 7ea0533
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion tests/python/mkl/test_mkldnn.py
Original file line number Diff line number Diff line change
Expand Up @@ -281,6 +281,7 @@ 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):
Expand All @@ -291,7 +292,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-6, rtol=0.16, atol=1e-4)
check_numeric_gradient(test, in_location, numeric_eps=1e-2, rtol=0.16, atol=1e-4)

stypes = ['row_sparse', 'default']
for stype in stypes:
Expand Down

0 comments on commit 7ea0533

Please sign in to comment.