fix xlnet & transfotests

2020-03-08 16:25:03 +01:00
parent 66c827656f
commit b4a3a64744
4 changed files with 24 additions and 64 deletions
--- a/tests/test_modeling_tf_xlnet.py
+++ b/tests/test_modeling_tf_xlnet.py
@@ -760,20 +760,10 @@ class TFXLNetModelLanguageGenerationTest(unittest.TestCase):
            9,
            4,
            3,
-            1722,
-            19,
-            24,
-            6348,
-            61,
-            977,
-            176,
-            1772,
-            33,
-            45,
-            970,
-            19,
-            4185,
            19,
+            12943,
+            4354,
+            153,
            27,
            442,
            22,