lucidrains/vit-pytorch: Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch
Category: ml-dl
Language: Python
Topics: artificial-intelligence, attention-mechanism, computer-vision, image-classification, transformers