Skip to content

GitLab

Unverified Commit 480a9d6b authored by abhishek thakur's avatar abhishek thakur Committed by GitHub
Browse files

Fix TFConvBertModelIntegrationTest::test_inference_masked_lm Test (#10104)

parent 0c3d23df
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
tests/test_modeling_tf_convbert.py
View file @ 480a9d6b
...@@ -389,9 +389,9 @@ class TFConvBertModelIntegrationTest(unittest.TestCase): ...@@ -389,9 +389,9 @@ class TFConvBertModelIntegrationTest(unittest.TestCase):
expected_slice = tf.constant( expected_slice = tf.constant(
[ [
[ [
[-0.03475493, -0.4686034, -0.30638832], [-0.10334751, -0.37152207, -0.2682219],
[0.22637248, -0.26988646, -0.7423424], [0.20078957, -0.3918426, -0.78811496],
[0.10324868, -0.45013508, -0.58280784], [0.08000169, -0.509474, -0.59314483],
] ]
] ]
) )
......
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