Skip to content

GitLab

Commit 7f2c0091 authored by LysandreJik's avatar LysandreJik
Browse files

TF BERT test update

parent e1b7e10d
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
tests/test_modeling_tf_bert.py
View file @ 7f2c0091
...@@ -356,9 +356,9 @@ class TFBertModelIntegrationTest(unittest.TestCase): ...@@ -356,9 +356,9 @@ class TFBertModelIntegrationTest(unittest.TestCase):
expected_slice = tf.constant( expected_slice = tf.constant(
[ [
[ [
[-0.05243197, -0.04498899, 0.05512108], [-0.05243197, -0.04498899, 0.05512108],
[-0.07444685, -0.01064632, 0.04352357], [-0.07444685, -0.01064632, 0.04352357],
[-0.05020351, 0.05530146, 0.00700043] [-0.05020351, 0.05530146, 0.00700043],
] ]
] ]
) )
......
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