di37/vision-transformers-from-scratch

A modular, from-scratch implementation of a Vision Transformer (ViT) in PyTorch, configurable for datasets.

语言：Jupyter Notebook · GitHub 星标：9 · Fork：0

在 GitHub 上查看原仓库