Fix expected loss values in some (m)T5 tests (#18177)

* fix expected loss values Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2022-07-18 15:26:21 +02:00
parent 7417f3acb7
commit cb19c2afdc
2 changed files with 4 additions and 4 deletions
--- a/tests/models/mt5/test_modeling_tf_mt5.py
+++ b/tests/models/mt5/test_modeling_tf_mt5.py
@@ -69,5 +69,5 @@ class TFMT5ModelIntegrationTest(unittest.TestCase):
        loss = model(input_ids, labels=labels).loss
        mtf_score = -tf.math.reduce_mean(loss).numpy()

-        EXPECTED_SCORE = -21.210594
+        EXPECTED_SCORE = -21.228168
        self.assertTrue(abs(mtf_score - EXPECTED_SCORE) < 2e-4)