github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

hotfix: adjust test_backward_pass_diamond_model thresholds (#5981)

Browse Source
This commit is contained in:
qazal
2024-08-09 00:20:53 +08:00
committed by GitHub
parent 08d22066ee
commit e6d41b0ce7
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_tensor.py
View File

@@ -141,7 +141,7 @@ class TestTinygrad(unittest.TestCase):
return out.detach().numpy(), u.grad, v.grad, w.grad
for x,y in zip(test_tinygrad(), test_pytorch()):
np.testing.assert_allclose(x, y, atol=1e-5)
np.testing.assert_allclose(x, y, atol=1e-5, rtol=1e-6)
def test_nograd(self):
x = Tensor(x_init, requires_grad=False)