search

lukemelas / PyTorch-Pretrained-ViT

Vision Transformer (ViT) in PyTorch
770 stars 124 forks source link

Will you release the L_16 pretrained model? #12

Open ljk1072911239 opened 3 years ago

ljk1072911239 commented 3 years ago

Will you release the L_16 pretrained model?

arkel23 commented 3 years ago

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 commented 3 years ago

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 commented 3 years ago

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