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):
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)
......@@ -661,7 +661,7 @@ class TFT5ModelIntegrationTests(unittest.TestCase):
loss = model(input_ids, labels=labels).loss
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)
@slow
......@@ -687,7 +687,7 @@ class TFT5ModelIntegrationTests(unittest.TestCase):
loss = model(input_ids, labels=labels).loss
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)
@slow
......@@ -711,7 +711,7 @@ class TFT5ModelIntegrationTests(unittest.TestCase):
loss = model(input_ids, labels=labels).loss
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)
@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