Skip to content

GitLab

Unverified Commit 82b62fa6 authored by Anton Lozhkov's avatar Anton Lozhkov Committed by GitHub
Browse files

Update SEW integration test tolerance (#14048)

parent bdf31d6e
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/test_modeling_sew.py
View file @ 82b62fa6
......@@ -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):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment