From 480a9d6ba053aa6906bad3575b6a1c793cc30497 Mon Sep 17 00:00:00 2001 From: abhishek thakur Date: Tue, 9 Feb 2021 20:22:54 +0100 Subject: [PATCH] Fix TFConvBertModelIntegrationTest::test_inference_masked_lm Test (#10104) --- tests/test_modeling_tf_convbert.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/test_modeling_tf_convbert.py b/tests/test_modeling_tf_convbert.py index 9c2a1b8104..9ba3e75b22 100644 --- 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], ] ] )