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

only mish failed

Browse Source
This commit is contained in:
George Hotz
2021-01-03 09:47:11 -08:00
parent a337f7780e
commit 8a38e0d207
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_ops.py
View File

@@ -75,7 +75,7 @@ class TestOps(unittest.TestCase):
def test_mish(self):
def _mish_pytorch(x):
return x*torch.tanh(torch.nn.functional.softplus(x))
helper_test_op([(45,65)], _mish_pytorch, Tensor.mish, atol=1e-6, grad_atol=1e-6)
helper_test_op([(45,65)], _mish_pytorch, Tensor.mish)
def test_dot(self):
helper_test_op([(45,65), (65,100)], lambda x,y: x.matmul(y), Tensor.dot)
def test_multidot(self):