copypaste tolerances

This commit is contained in:
justheuristic 2022-09-17 23:41:40 +03:00
parent eac9aca460
commit 7facedda38

View File

@ -427,4 +427,4 @@ def test_matmullt(
) )
if req_grad[2]: if req_grad[2]:
torch.testing.assert_allclose(gradBias1, gradBias2) torch.testing.assert_allclose(gradBias1, gradBias2, atol=0.18, rtol=0.3)