-
Notifications
You must be signed in to change notification settings - Fork 716
Transfer Learning Guide #2394
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
Transfer Learning Guide #2394
Conversation
Check out this pull request on See visual diffs & provide feedback on Jupyter Notebooks. Powered by ReviewNB |
Codecov Report
@@ Coverage Diff @@
## main #2394 +/- ##
==========================================
- Coverage 79.47% 79.46% -0.01%
==========================================
Files 49 49
Lines 5204 5202 -2
==========================================
- Hits 4136 4134 -2
Misses 1068 1068
Help us with your feedback. Take ten seconds to tell us how you rate us. Have a feature suggestion? Share it here. |
c256185
to
95f98b3
Compare
34a9fb0
to
255cc7a
Compare
3ee1daa
to
58eb73d
Compare
21ecc39
to
aea323b
Compare
cb98b14
to
2993338
Compare
Hey @jheek, using |
8f89edf
to
3200c07
Compare
@8bitmp3 thanks a lot for the feedback! It was really useful, I've made the stylistic changes and will keep them in mind for future guides. |
359f962
to
76f052f
Compare
8250e0b
to
6b8016f
Compare
Reviewed and updated in cgarciae#1. LMKWYT! |
ac2ebde
to
2047050
Compare
c882af3
to
cac954e
Compare
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.
Thank you @cgarciae
@jheek @andsteing @levskaya @IvyZX If you're happy with our changes, please approve/LGTM. Thanks!
What does this PR do?
Adds a Transfer Learning guide which includes:
optax.multi_transform
Live preview: https://flax--2394.org.readthedocs.build/en/2394/guides/transfer_learning.html
Notes
vit_jax
becausejax
versions conflict, so I used a model from HuggingFace'stransformers
library. Only quirk here is thattransformers
downgradesjax
so a fix is added to upgrade it on CI.