适用于 Linux 的 Vision Transformer PyTorch 下载

这是一个名为 Vision Transformer Pytorch 的 Linux 应用程序，其最新版本可以下载，文件名为 1.15.2sourcecode.tar.gz。它可以在免费主机提供商 OnWorks 的工作站服务器上在线运行。

SCREENSHOTS

Vision Transformer PyTorch

商品描述

本仓库提供了一个基于 PyTorch 的极简视觉 Transformer (ViT) 实现，专注于图像分类所需的核心架构组件。它将模型分解为图像块嵌入、位置编码、多头自注意力机制、前馈模块和分类头，以便您可以单独理解每个组件。代码刻意保持简洁和模块化，方便您调整超参数、深度、宽度和注意力维度。由于它与原生 PyTorch 非常接近，您可以集成自定义数据集和训练循环，而无需担心框架锁定。它被广泛用作学习视觉 Transformer 模型的参考资料，以及研究原型所需的轻量级基线。本项目鼓励进行实验——您可以更换优化器、更改数据增强方法，或将 Transformer 主干网络集成到下游任务中。

功能

• 简洁的 PyTorch 模块，用于补丁、注意力机制、MLP 模块和头部
• 可轻松配置深度、喷头、尺寸和漏水设置
• 可直接插入常用循环的简单训练和推理示例
• 便于基于自定义数据进行实验和快速原型开发
• 最小的外部依赖和惯用的 PyTorch 风格
• 可作为 ViT 架构细节的易读参考资料。

程式语言

Python

分类

您也可以从 https://sourceforge.net/projects/vision-tran-pytorch.mirror/ 获取此应用程序。它已托管在 OnWorks 上，以便通过我们的免费操作系统之一以最简单的方式在线运行。