vision-transformer-pytorch

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

Description

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch 在PyTorch中实现视觉Transformer，这是一种仅通过单个Transformer编码器就在视觉分类任务中达到最先进水平的简单方法。

artificial-intelligence attention-mechanism computer-vision image-classification transformers

Readme MIT 11 MiB