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

Commit

Permalink
Fix flaky unit test layer_norm
Browse files Browse the repository at this point in the history
  • Loading branch information
reminisce committed Mar 23, 2018
1 parent 6431fb4 commit ecc8466
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/python/unittest/test_operator.py
Original file line number Diff line number Diff line change
Expand Up @@ -2539,7 +2539,7 @@ def npy_layer_norm(data, gamma, beta, axis=1, eps=1E-5):
for req in ['write', 'add']:
check_numeric_gradient(out_s, {'data': data, 'gamma': gamma, 'beta': beta},
grad_nodes={'data': req, 'gamma': req, 'beta': req},
numeric_eps=1e-2, rtol=1e-2, atol=1e-3)
numeric_eps=1e-2, rtol=1e-2, atol=1e-2)

def test_layer_norm():
for dtype, forward_check_eps in zip([np.float16, np.float32, np.float64],
Expand Down

0 comments on commit ecc8466

Please sign in to comment.