-
Notifications
You must be signed in to change notification settings - Fork 27.7k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
implementing tflxmertmodel integration test #12497
Conversation
I believe your version of the code quality tools isn't up to date - can you try running the following at the root of your clone: |
Hi @LysandreJik Yep It was my black version |
tests/test_modeling_tf_lxmert.py
Outdated
] | ||
] | ||
) | ||
tf.debugging.asset_near(output[:, :3, :3], expected_slice, atol=1e-4) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This should be assert_near
tf.debugging.asset_near(output[:, :3, :3], expected_slice, atol=1e-4) | |
tf.debugging.assert_near(output[:, :3, :3], expected_slice, atol=1e-4) |
tests/test_modeling_tf_lxmert.py
Outdated
[self.model_tester.num_attention_heads, encoder_key_length, self.model_tester.num_visual_features], | ||
[ | ||
self.model_tester.num_attention_heads, | ||
encoder_key_length, | ||
self.model_tester.num_visual_features, | ||
], |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
There's a lot of style changes in your PR - do you think you could only do the changes relative to the code you're adding? Thank you!
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Great, thanks @sadakmed!
What does this PR do?
this PR implement an integration test for TFLxmertmodel as requested in #9954
@LysandreJik