-
Notifications
You must be signed in to change notification settings - Fork 812
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 projection layer to roberta encoder #1451
base: main
Are you sure you want to change the base?
Conversation
Codecov Report
@@ Coverage Diff @@
## main #1451 +/- ##
==========================================
- Coverage 86.38% 86.01% -0.38%
==========================================
Files 58 58
Lines 2226 2245 +19
==========================================
+ Hits 1923 1931 +8
- Misses 303 314 +11
Continue to review full report at Codecov.
|
Overall LGTM. It seems like we're getting a lot of code coverage warnings. Should we also add unit tests for the |
Yes, I think it's a good idea. The models tests didn't really touch this part of the code since we are not projecting the encoder output for standard architectures. Though, it would be more like integration/smoke tests instead of unit-test. We also do not have unit tests for other modules as well, but they are covered well through our end-2-end model tests (we call them unit-tests for now, but as @mthrok commented in here, we are going to move them to integration tests as well). |
Gotcha this makes sense. We can have some followup tasks to add these integrations tests. Stamped! |
User may want to additionally project the features from encoder. This PR add support for projecting features to different dimensional space.