vision-transformer-pytorch icon indicating copy to clipboard operation
vision-transformer-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon asyml

R50+ViT-B/16 hybrid model ?

Open GuHongyang opened this issue 4 years ago • 2 comments

I found that Google has updated the R50+ViT-B/16 hybrid model. Can the pytorch version also be updated?

GuHongyang avatar Dec 10 '20 05:12 GuHongyang

We have provided the PyTorch version of the code and pre-trained model in https://github.com/tczhangzhi/VisionTransformer-Pytorch. If the authors have no other plans, we are glad to make PR to this repo. You can currently use R50+ViT-B in the following ways:

from vision_transformer_pytorch import VisionTransformer

net = VisionTransformer.from_pretrained('R50+ViT-B_16')

tczhangzhi avatar Feb 16 '21 03:02 tczhangzhi

Thank you! Could you make a PR to the repo?

christy-yuan-li avatar Feb 23 '21 09:02 christy-yuan-li