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

the provided pretrained weight only includes the encoder weight, not includes the decoder weight #8

Closed
wangqingbin opened this issue Jan 10, 2022 · 2 comments

Comments

@wangqingbin
Copy link

It seems that the provided pretrained weight only includes the encoder weight, not includes the decoder weight, could you provide the entire weight for pretraining model. Thank you very much!

@KaimingHe
Copy link
Contributor

See https://dl.fbaipublicfiles.com/mae/pretrain/mae_pretrain_vit_base_full.pth. The other two files mae_pretrain_vit_large_full.pth and mae_pretrain_vit_huge_full.pth are in the same dir too. Note these models reconstruct normalized pixels.

@tristanengst
Copy link

Could you confirm the hyperparameters used for training the model linked above? I need to train them for a few more epochs so as to compare with a related method I'm working on. Thanks so much!

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

3 participants