update rtol

Lightning-AI · Jul 27, 2020 · 4320329 · 4320329
1 parent 5ae0ef5
commit 4320329
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/models/test_grad_norm.py b/tests/models/test_grad_norm.py
@@ -43,7 +43,7 @@ def on_after_backward(self):
 
 
 @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