Skip to content

GitLab

Unverified Commit ea46e3fa authored by Lysandre Debut's avatar Lysandre Debut Committed by GitHub
Browse files

Adjust loss difference (#10669)

parent c526bde3
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/test_modeling_tf_mt5.py
View file @ ea46e3fa
...@@ -53,4 +53,4 @@ class TFMT5ModelIntegrationTest(unittest.TestCase): ...@@ -53,4 +53,4 @@ class TFMT5ModelIntegrationTest(unittest.TestCase):
mtf_score = -tf.math.reduce_sum(loss).numpy() mtf_score = -tf.math.reduce_sum(loss).numpy()
EXPECTED_SCORE = -84.9127 EXPECTED_SCORE = -84.9127
self.assertTrue(abs(mtf_score - EXPECTED_SCORE) < 1e-4) self.assertTrue(abs(mtf_score - EXPECTED_SCORE) < 2e-4)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment