diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/test_autograd.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/test_autograd.py b/tests/test_autograd.py index 5171c4f..28d9259 100644 --- a/tests/test_autograd.py +++ b/tests/test_autograd.py @@ -427,4 +427,4 @@ def test_matmullt( ) if req_grad[2]: - torch.testing.assert_allclose(gradBias1, gradBias2) + torch.testing.assert_allclose(gradBias1, gradBias2, atol=0.18, rtol=0.3) |