-
Notifications
You must be signed in to change notification settings - Fork 31.2k
Add integration tests lm generate torch tf #3191
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
Add integration tests lm generate torch tf #3191
Conversation
|
Approx. how long do these tests take on a single V100? |
Don't know how long they take on a V100. On a cpu, all tests combined (7 model tests for PT and 6 model tests for TF) take less than 10min (whereas |
Codecov Report
@@ Coverage Diff @@
## master #3191 +/- ##
==========================================
+ Coverage 78.15% 78.16% +0.01%
==========================================
Files 98 98
Lines 16641 16641
==========================================
+ Hits 13006 13008 +2
+ Misses 3635 3633 -2
Continue to review full report at Codecov.
|
LysandreJik
left a comment
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.
Nice, LGTM!
|
|
||
| tensor = inputs_embeds + self.position_embeddings(position_ids) | ||
| if langs is not None and self.use_lang_emb: | ||
| if langs is not None and self.use_lang_emb and self.n_langs > 1: |
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.
Nice catch!
|
Good to merge for me |
Add integration tests for all LM models that are able to generate language.
do_sample=False(greedy) generation and verify that TF 2.0 and PT yield the same results.