commaai
/
tinygrad
mirror of https://github.com/commaai/tinygrad.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

increase atol on reset train

This commit is contained in:
George Hotz 2024-03-24 15:17:31 -07:00
parent d8fafca13a
commit 03899a74bb
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_multitensor.py
View File

@ -261,7 +261,7 @@ class TestMultiTensor(unittest.TestCase):
shard_output.backward()
shard_grad = m.conv1.weight.grad.numpy()
# sometimes there is zeros in these grads... why?
np.testing.assert_allclose(grad, shard_grad, atol=1e-6, rtol=1e-6)
np.testing.assert_allclose(grad, shard_grad, atol=3e-6, rtol=3e-6)
def test_multi_tensor_jit_param(self):
@TinyJit