From 4320329540798c112cf45dcbd6f677993e4c6ad6 Mon Sep 17 00:00:00 2001 From: rohitgr7 Date: Tue, 28 Jul 2020 02:34:17 +0530 Subject: [PATCH] update rtol --- tests/models/test_grad_norm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/models/test_grad_norm.py b/tests/models/test_grad_norm.py index d7978965a3..f8da7730b8 100644 --- 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