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
[Feature]: Support BEiT Transformer layer #919
Conversation
Codecov Report
@@ Coverage Diff @@
## dev-1.x #919 +/- ##
==========================================
- Coverage 0.02% 0.02% -0.01%
==========================================
Files 121 121
Lines 8217 8334 +117
Branches 1368 1382 +14
==========================================
Hits 2 2
- Misses 8215 8332 +117
Flags with carried forward coverage won't be shown. Click here to find out more.
Help us with your feedback. Take ten seconds to tell us how you rate us. Have a feature suggestion? Share it here. |
@@ -98,6 +101,109 @@ def forward(self, x): | |||
return x | |||
|
|||
|
|||
class BEiTTransformerEncoderLayer(TransformerEncoderLayer): |
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.
Please add unit tests.
* [Feature]: Add BEiT-style transformer encoder layer * [Feature]: Add average token * [Fix]: Fix lint * [Fix]: Refactor CAE config * [Fix]: Change cv2 backend to pillow backend * [Fix]: Fix MAE and CAE reshape bug * [Feature]: Add freeze vit layers * [Feature]: Add mc * [Fix]: Fix lint * [Fix]: Fix dataset bug * [Fix]: Delete cae selfsup config * [Fix]: docstring * [Refactor]: Add init_values to layer_scalue_init_value * [Fix]: Refine the docstring of avg_token * [Fix]: Call super init weight in beit attention * [Fix]: remove mc * [Fix]: Fix docstring * [Fix]: Fix docstring * [Fix]: Fix lint * [Fix]: Fix init_value bug and change the logic of outputting cls token * [Fix]: Fix docstring
Thanks for your contribution and we appreciate it a lot. The following instructions would make your pull request more healthy and more easily get feedback. If you do not understand some items, don't worry, just make the pull request and seek help from maintainers.
Motivation
Please describe the motivation of this PR and the goal you want to achieve through this PR.
Modification
Please briefly describe what modification is made in this PR.
BC-breaking (Optional)
Does the modification introduce changes that break the backward compatibility of the downstream repositories?
If so, please describe how it breaks the compatibility and how the downstream projects should modify their code to keep compatibility with this PR.
Use cases (Optional)
If this PR introduces a new feature, it is better to list some use cases here and update the documentation.
Checklist
Before PR:
After PR: