Fix TFConvBertModelIntegrationTest::test_inference_masked_lm Test (#10104)

480a9d6b · abhishek thakur · GitHub · 0c3d23df · 480a9d6b
Unverified Commit 480a9d6b authored Feb 09, 2021 by abhishek thakur Committed by GitHub Feb 09, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

tests/test_modeling_tf_convbert.py tests/test_modeling_tf_convbert.py +3 -3

No files found.
--- a/tests/test_modeling_tf_convbert.py
+++ b/tests/test_modeling_tf_convbert.py
@@ -389,9 +389,9 @@ class TFConvBertModelIntegrationTest(unittest.TestCase):
        expected_slice = tf.constant(
            [
                [
-                    [-0.03475493, -0.4686034, -0.30638832],
-                    [0.22637248, -0.26988646, -0.7423424],
-                    [0.10324868, -0.45013508, -0.58280784],
+                    [-0.10334751, -0.37152207, -0.2682219],
+                    [0.20078957, -0.3918426, -0.78811496],
+                    [0.08000169, -0.509474, -0.59314483],
                ]
            ]
        )