Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
1f5d9513
Commit
1f5d9513
authored
Oct 11, 2019
by
thomwolf
Browse files
fix test
parent
0f9fc4fb
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
5 additions
and
2 deletions
+5
-2
transformers/tests/modeling_tf_xlnet_test.py
transformers/tests/modeling_tf_xlnet_test.py
+2
-1
transformers/tests/modeling_xlnet_test.py
transformers/tests/modeling_xlnet_test.py
+3
-1
No files found.
transformers/tests/modeling_tf_xlnet_test.py
View file @
1f5d9513
...
@@ -161,7 +161,8 @@ class TFXLNetModelTest(TFCommonTestCases.TFCommonModelTester):
...
@@ -161,7 +161,8 @@ class TFXLNetModelTest(TFCommonTestCases.TFCommonModelTester):
"outputs"
:
outputs
.
numpy
(),
"outputs"
:
outputs
.
numpy
(),
}
}
model
.
config
.
mem_len
=
0
config
.
mem_len
=
0
model
=
TFXLNetModel
(
config
)
no_mems_outputs
=
model
(
inputs
)
no_mems_outputs
=
model
(
inputs
)
self
.
parent
.
assertEqual
(
len
(
no_mems_outputs
),
1
)
self
.
parent
.
assertEqual
(
len
(
no_mems_outputs
),
1
)
...
...
transformers/tests/modeling_xlnet_test.py
View file @
1f5d9513
...
@@ -150,7 +150,9 @@ class XLNetModelTest(CommonTestCases.CommonModelTester):
...
@@ -150,7 +150,9 @@ class XLNetModelTest(CommonTestCases.CommonModelTester):
"outputs"
:
outputs
,
"outputs"
:
outputs
,
}
}
model
.
config
.
mem_len
=
0
config
.
mem_len
=
0
model
=
XLNetModel
(
config
)
model
.
eval
()
no_mems_outputs
=
model
(
input_ids_1
)
no_mems_outputs
=
model
(
input_ids_1
)
self
.
parent
.
assertEqual
(
len
(
no_mems_outputs
),
1
)
self
.
parent
.
assertEqual
(
len
(
no_mems_outputs
),
1
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment