SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Update SEW integration test tolerance (#14048)

Browse Source
This commit is contained in:
Anton Lozhkov
2021-10-18 13:58:59 +03:00
committed by GitHub
parent bdf31d6e0a
commit 82b62fa607
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_modeling_sew.py
View File

@@ -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_first, atol=5e-3))
self.assertTrue(torch.allclose(outputs[:, -4:, -4:], expected_outputs_last, 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 @tooslow
def test_inference_ctc_batched(self): def test_inference_ctc_batched(self):