dnrops
/
transformers
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix TFConvBertModelIntegrationTest::test_inference_masked_lm Test (#10104)

This commit is contained in:
abhishek thakur 2021-02-09 20:22:54 +01:00 committed by GitHub
parent 0c3d23dff7
commit 480a9d6ba0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/test_modeling_tf_convbert.py
View File

@ -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],
]
]
)