文件名称:VisionTransformer-Pytorch
文件大小:25KB
文件格式:ZIP
更新时间:2024-06-03 06:53:13
Python
视觉变压器火炬 此项目是从 / 和 / ,为您提供了现成的API,可让您像EfficientNet一样容易地使用VisionTransformer。 快速开始 使用pip install vision_transformer_pytorch安装,并使用以下命令加载经过预训练的VisionTransformer: from vision_transformer_pytorch import VisionTransformer model = VisionTransformer.from_pretrained('ViT-B_16') 关于视觉变压器PyTorch Vision Transformer Pytorch是Vision Transformer的PyTorch重新实现,它基于常用的深度学习库的最佳实践之一 ,以及的优雅实现工具 。 在这个项目中,我们旨在使我们的PyTorch实施尽
【文件预览】:
VisionTransformer-Pytorch-main
----test.py(150B)
----examples()
--------imagenet()
----jax_to_pytorch()
--------convert_jax_to_pt()
--------README.md(1KB)
----LICENSE(11KB)
----setup.py(4KB)
----README.md(3KB)
----tests()
--------test_model.py(2KB)
----.gitignore(1KB)
----vision_transformer_pytorch()
--------utils.py(6KB)
--------model.py(12KB)
--------__init__.py(153B)
--------resnet.py(6KB)