没有合适的资源?快使用搜索试试~ 我知道了~
vit-pytorch:在Pytorch中实现视觉变压器,这是仅使用一个变压器编码器即可在视觉分类中实现SOTA的简单方法
共15个文件
py:7个
png:2个
license:1个
需积分: 45 7 下载量 95 浏览量
2021-03-16
09:05:05
上传
评论
收藏 5.66MB ZIP 举报
温馨提示
视觉变压器-火炬 实现,这是在Pytorch中仅使用一个变压器编码器即可在视觉分类中实现SOTA的一种简单方法。 视频中进一步解释了。 此处实际上没有多少代码,但也可以为所有人进行布局,因此我们可以加快注意力的革命。 有关使用预训练模型的Pytorch实施,请查看Ross Wightman的存储库。 官方的Jax存储库在。 安装 $ pip install vit-pytorch 用法 import torch from vit_pytorch import ViT v = ViT ( image_size = 256 , patch_size = 32 , num_classes = 1000 , dim = 1024 , depth = 6 , heads = 16 , mlp_dim = 2048 , dropout
资源推荐
资源详情
资源评论
收起资源包目录
vit-pytorch-main.zip (15个子文件)
vit-pytorch-main
vit_pytorch
vit_pytorch.py 4KB
t2t.py 3KB
distill.py 4KB
__init__.py 40B
mpp.py 6KB
efficient.py 2KB
.github
workflows
python-publish.yml 865B
t2t.png 109KB
LICENSE 1KB
examples
cats_and_dogs.ipynb 1.06MB
vit.gif 5.76MB
setup.py 757B
.gitignore 2KB
README.md 11KB
distill.png 49KB
共 15 条
- 1
资源评论
司幽幽
- 粉丝: 30
- 资源: 4547
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功