update rtol

2020-07-28 02:34:17 +05:30 · 2020-07-28 02:34:17 +05:30 · 4320329540
parent 5ae0ef5268
commit 4320329540
1 changed files with 1 additions and 1 deletions
--- a/tests/models/test_grad_norm.py
+++ b/tests/models/test_grad_norm.py
@ -43,7 +43,7 @@ class ModelWithManualGradTracker(EvalModelTemplate):


@pytest.mark.parametrize("norm_type", [1., 1.25, 1.5, 2, 3, 5, 10, 'inf'])
-def test_grad_tracking(tmpdir, norm_type, rtol=5e-3):
+def test_grad_tracking(tmpdir, norm_type, rtol=1e-3):
    os.environ['PL_DEV_DEBUG'] = '1'

    # rtol=5e-3 respects the 3 decimals rounding in `.grad_norms` and above