vit-pytorch

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

Transformers

同类项目

精选

Google 的端到端机器学习平台，生态完整。

精选

数万预训练模型的统一接口，覆盖文本、图像、音频与多模态。

精选

动态计算图的深度学习框架，研究与工业界的主流选择。

精选

Python 经典机器学习库，简单高效。

多后端（JAX / TF / PyTorch）的深度学习 API。

最简单快速的中型 GPT 训练与微调仓库。