SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

TF BERT test update

Browse Source
This commit is contained in:
LysandreJik
2020-11-23 18:19:54 -05:00
parent e1b7e10d5f
commit 7f2c00913a
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/test_modeling_tf_bert.py
View File

@@ -356,9 +356,9 @@ class TFBertModelIntegrationTest(unittest.TestCase):
expected_slice = tf.constant(
[
[
[-0.05243197, -0.04498899, 0.05512108],
[-0.07444685, -0.01064632, 0.04352357],
[-0.05020351, 0.05530146, 0.00700043]
[-0.05243197, -0.04498899, 0.05512108],
[-0.07444685, -0.01064632, 0.04352357],
[-0.05020351, 0.05530146, 0.00700043],
]
]
)