lucidrains / vit-pytorch

git clone https://github.com/lucidrains/vit-pytorch.git

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch