Update SEW integration test tolerance (#14048)

2021-10-18 13:58:59 +03:00
parent bdf31d6e0a
commit 82b62fa607
1 changed files with 1 additions and 1 deletions
--- a/tests/test_modeling_sew.py
+++ b/tests/test_modeling_sew.py
@@ -467,7 +467,7 @@ class SEWModelIntegrationTest(unittest.TestCase):

        self.assertTrue(torch.allclose(outputs[:, :4, :4], expected_outputs_first, atol=5e-3))
        self.assertTrue(torch.allclose(outputs[:, -4:, -4:], expected_outputs_last, atol=5e-3))
-        self.assertTrue(abs(outputs.sum() - expected_output_sum) < 2)
+        self.assertTrue(abs(outputs.sum() - expected_output_sum) < 5)

    @tooslow
    def test_inference_ctc_batched(self):