lucidrains/vit-pytorch

Whether to include pre-trained models

Closed this issue · 1 comments

Thanks for such a great job

does the repository include pre-trained models, which I am not seeing at the moment, or should I set up my own dataset to start training based on the implemented code?

Hi @KawaiiAsh , were you able to find pretrained weights for the implementation in this repo? Or are we to look into https://github.com/rwightman/pytorch-image-models