Adjust loss difference (#10669)

2021-03-12 01:09:46 -05:00
parent c526bde319
commit ea46e3fa9c
1 changed files with 1 additions and 1 deletions
--- a/tests/test_modeling_tf_mt5.py
+++ b/tests/test_modeling_tf_mt5.py
@@ -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)