mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-31 02:02:21 +06:00
Adjust loss difference (#10669)
This commit is contained in:
parent
c526bde319
commit
ea46e3fa9c
@ -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)
|
||||
|
Loading…
Reference in New Issue
Block a user