From 82b62fa60771bc634d864be874eaad8297c11beb Mon Sep 17 00:00:00 2001 From: Anton Lozhkov Date: Mon, 18 Oct 2021 13:58:59 +0300 Subject: [PATCH] Update SEW integration test tolerance (#14048) --- tests/test_modeling_sew.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_modeling_sew.py b/tests/test_modeling_sew.py index 45556585eb..4cf7ce5a41 100644 --- 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):