Skip to content
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

[WIP] Add SwinTransformerV2 (CVPR'22) #2986

Open
wants to merge 9 commits into
base: dev-1.x
Choose a base branch
from

Conversation

liu-mengyang
Copy link
Contributor

Motivation

Add SwinTransformerV2 to resolve #1778 .

Modification

TBD.

BC-breaking (Optional)

TBD.

Use cases (Optional)

TBD.

Checklist

  • Pre-commit or other linting tools are used to fix the potential lint issues.
  • The modification is covered by complete unit tests. If not, please add more unit test to ensure the correctness.
  • If the modification has potential influence on downstream projects, this PR should be tested with downstream projects, like MMDet or MMDet3D.
  • The documentation has been modified accordingly, like docstring or example tutorials.

Acknowledgement

TBD.

@xiexinch
Copy link
Collaborator

Hi @liu-mengyang,
Thanks for your contribution!
Is there any update?

@liu-mengyang
Copy link
Contributor Author

Hi, too many things these two weeks... Tonight I will push 1 updated version.

@xiexinch
Copy link
Collaborator

Hi @liu-mengyang,
Could you provide the converted weights for inference?

@liu-mengyang
Copy link
Contributor Author

Hi, I am still working on it :)

Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

We might add more information to README.md.

float(-100.0)).masked_fill(
attn_mask == 0, float(0.0))
else:
shifted_query = x
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Can't find the definition of x

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I fixed it and I am trying to train the model on my personal server.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

3 participants