lucidrains/vit-pytorch
Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch
Language: Python
#artificial_intelligence #attention_mechanism #computer_vision #image_classification #transformers
Stars: 242 Issues: 1 Forks: 15
https://github.com/lucidrains/vit-pytorch