Open ljk1072911239 opened 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
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.
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
Will you release the L_16 pretrained model?