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

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)