bottleneck-transformer-pytorch:Pytorch中瓶颈变压器的实现

时间:2024-03-25 21:14:30
【文件属性】:

文件名称:bottleneck-transformer-pytorch:Pytorch中瓶颈变压器的实现

文件大小:82KB

文件格式:ZIP

更新时间:2024-03-25 21:14:30

deep-learning transformers artificial-intelligence vision image-classification

瓶颈变压器-火炬 在Pytorch中,在性能-计算权衡方面优于EfficientNet和DeiT的卷积(SotA)视觉识别模型的卷积+注意实现 安装 $ pip install bottleneck-transformer-pytorch 用法 import torch from torch import nn from bottleneck_transformer_pytorch import BottleStack layer = BottleStack ( dim = 256 , # channels in fmap_size = 64 , # feature map size dim_out = 2048 , # channels out proj_factor = 4 , #


【文件预览】:
bottleneck-transformer-pytorch-main
----.github()
--------workflows()
----bottle-diagram-2.png(46KB)
----bottleneck_transformer_pytorch()
--------bottleneck_transformer_pytorch.py(7KB)
--------__init__.py(99B)
----LICENSE(1KB)
----setup.py(810B)
----README.md(2KB)
----.gitignore(2KB)
----bottle-diagram.png(29KB)

网友评论