Adjust loss difference (#10669)

This commit is contained in:
Lysandre Debut 2021-03-12 01:09:46 -05:00 committed by GitHub
parent c526bde319
commit ea46e3fa9c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -53,4 +53,4 @@ class TFMT5ModelIntegrationTest(unittest.TestCase):
mtf_score = -tf.math.reduce_sum(loss).numpy()
EXPECTED_SCORE = -84.9127
self.assertTrue(abs(mtf_score - EXPECTED_SCORE) < 1e-4)
self.assertTrue(abs(mtf_score - EXPECTED_SCORE) < 2e-4)