Skip to content

Commit

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

This reverts commit 445967e.
  • Loading branch information
lebeg committed Sep 11, 2018
1 parent 597a637 commit 5b1b952
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 5b1b952

Please sign in to comment.