Unverified Commit cb19c2af authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

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



* fix expected loss values
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent 7417f3ac
...@@ -69,5 +69,5 @@ class TFMT5ModelIntegrationTest(unittest.TestCase): ...@@ -69,5 +69,5 @@ class TFMT5ModelIntegrationTest(unittest.TestCase):
loss = model(input_ids, labels=labels).loss loss = model(input_ids, labels=labels).loss
mtf_score = -tf.math.reduce_mean(loss).numpy() 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) self.assertTrue(abs(mtf_score - EXPECTED_SCORE) < 2e-4)
...@@ -661,7 +661,7 @@ class TFT5ModelIntegrationTests(unittest.TestCase): ...@@ -661,7 +661,7 @@ class TFT5ModelIntegrationTests(unittest.TestCase):
loss = model(input_ids, labels=labels).loss loss = model(input_ids, labels=labels).loss
mtf_score = -tf.math.reduce_mean(loss).numpy() mtf_score = -tf.math.reduce_mean(loss).numpy()
EXPECTED_SCORE = -4.7710114 EXPECTED_SCORE = -4.771147
self.assertTrue(abs(mtf_score - EXPECTED_SCORE) < 1e-4) self.assertTrue(abs(mtf_score - EXPECTED_SCORE) < 1e-4)
@slow @slow
...@@ -687,7 +687,7 @@ class TFT5ModelIntegrationTests(unittest.TestCase): ...@@ -687,7 +687,7 @@ class TFT5ModelIntegrationTests(unittest.TestCase):
loss = model(input_ids, labels=labels).loss loss = model(input_ids, labels=labels).loss
mtf_score = -tf.math.reduce_mean(loss).numpy() mtf_score = -tf.math.reduce_mean(loss).numpy()
EXPECTED_SCORE = -14.759922 EXPECTED_SCORE = -14.757326
self.assertTrue(abs(mtf_score - EXPECTED_SCORE) < 1e-4) self.assertTrue(abs(mtf_score - EXPECTED_SCORE) < 1e-4)
@slow @slow
...@@ -711,7 +711,7 @@ class TFT5ModelIntegrationTests(unittest.TestCase): ...@@ -711,7 +711,7 @@ class TFT5ModelIntegrationTests(unittest.TestCase):
loss = model(input_ids, labels=labels).loss loss = model(input_ids, labels=labels).loss
mtf_score = -tf.math.reduce_mean(loss).numpy() mtf_score = -tf.math.reduce_mean(loss).numpy()
EXPECTED_SCORE = -7.594554 EXPECTED_SCORE = -7.592465
self.assertTrue(abs(mtf_score - EXPECTED_SCORE) < 1e-4) self.assertTrue(abs(mtf_score - EXPECTED_SCORE) < 1e-4)
@slow @slow
......
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