From ea46e3fa9c0fce752bdf6e73c9e0f10a8d04ae37 Mon Sep 17 00:00:00 2001 From: Lysandre Debut Date: Fri, 12 Mar 2021 01:09:46 -0500 Subject: [PATCH] Adjust loss difference (#10669) --- tests/test_modeling_tf_mt5.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_modeling_tf_mt5.py b/tests/test_modeling_tf_mt5.py index 422e344075..9b23e05f75 100644 --- 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)