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

Will you release the L_16 pretrained model? #12

Open
ljk1072911239 opened this issue Feb 26, 2021 · 3 comments
Open

Will you release the L_16 pretrained model? #12

ljk1072911239 opened this issue Feb 26, 2021 · 3 comments

Comments

@ljk1072911239
Copy link

Will you release the L_16 pretrained model?

@arkel23
Copy link

arkel23 commented Mar 18, 2021

I added a few lines to download the L_16 pretrained model (and the others not included in here) and make it compatible with the rest of this implementation:
https://github.com/arkel23/PyTorch-Pretrained-ViT

@ljk1072911239
Copy link
Author

I added a few lines to download the L_16 pretrained model (and the others not included in here) and make it compatible with the rest of this implementation:
https://github.com/arkel23/PyTorch-Pretrained-ViT

How can I change the npz suffix file to pth suffix file? Thank you.

@arkel23
Copy link

arkel23 commented Mar 22, 2021

Just run the download_convert_models.py script with the models you would like to download and convert to .pth
You can modify the list in that file models_list = ['B_16', 'B_32', 'L_16', 'L_32', 'H_14', 'B_16_imagenet1k', 'B_32_imagenet1k', 'L_16_imagenet1k', 'L_32_imagenet1k'] if you don't want to download certain models. It will save the converted files to the default directory for pytorch models which depends on your operating system
https://github.com/arkel23/PyTorch-Pretrained-ViT/blob/master/download_convert_models.py

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