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

Pretraining question #4

Open
noahcjones opened this issue Dec 8, 2023 · 2 comments
Open

Pretraining question #4

noahcjones opened this issue Dec 8, 2023 · 2 comments

Comments

@noahcjones
Copy link

Hi, does the SwinTransformer v2 do SimMiM pretraining? This is shown in the paper:
https://arxiv.org/pdf/2111.09883.pdf

If not, any plans to add/how difficult would it be to port?

@shkarupa-alex
Copy link
Owner

No, provided weights converted from original V1 and V2 models trained on ImageNet (22k when available, 1k otherwise).
If you want to try simmim weights you can modify this script https://github.com/shkarupa-alex/tfswin/blob/master/convert_weights.py to port required weights

@noahcjones
Copy link
Author

Oh okay, I ask because I'd like to pretrain on my own, not use the converted weights. I see the example here vit-tensorflow and would like to do something similar with the swinv2.

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

No branches or pull requests

2 participants