-
Notifications
You must be signed in to change notification settings - Fork 423
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 NLP Algorithms Tests #1839
Add NLP Algorithms Tests #1839
Conversation
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.
Mostly looks good, thanks! Left a bunch of minor comments. Also, for any of the follow up work you mention in the PR description, if you could make a JIRA and link it that would be great. Lastly, I think we still need to add some stuff for Alibi and SeqLengthWarmup in tests/algorithms/algorithm_settings.py
. These are used for the tests that test all algorithms. Maybe this is coming in your export PR/commits though
…nto nikhil/nlp-testing
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.
LGTM, left a couple small comments. Thanks!
What does this PR do?
This PR updates many algorithms tests to run on NLP models, as guided by our NLP testing plan: https://www.notion.so/NLP-74e31fb2d8f0472e9fcd6fd98dd67686
The testing plan states that all algorithms that support NLP models should have tests that run on a HuggingFace BERT model and a Simple Transformer model.
Note: Tests for Stochastic Weight Averaging do not exist, but should be implemented by the original author. That is outside the scope of this PR (JIRA: CO-1611)
All other algorithms NLP algorithms are included in this PR. However,
test_factorize_algorithm.py
fails on some SimpleTransformer model tests. The Factorize algorithm needs to be improved to account for this. Included in this PR is updated documentation making clear its shortcoming (JIRA: CO-1612)Additional PRs will include tests for exporting NLP models.