【文件属性】：
文件名称：变压器中的变压器：Pytorch中的Transformer在Transformer中的实现，像素级别注意与补丁级别注意配对以进行图像分类
文件大小：76KB
文件格式：ZIP
更新时间：2021-03-03 11:47:58
deep-learning transformers artificial-intelligence image-classification Python
变压器中的变压器
Pytorch中实现，像素级别注意与补丁级别注意配对以进行图像分类。
安装
$ pip install transformer-in-transformer
用法
import torch
from transformer_in_transformer import TNT
tnt = TNT (
image_size = 256 , # size of image
patch_dim = 512 , # dimension of patch token
pixel_dim = 24 , # dimension of pixel token
patch_size = 16 , # patch size
pixel_size = 4 , # pixel size
【文件预览】：
transformer-in-transformer-main
----setup.py(785B)
----.gitignore(2KB)
----LICENSE(1KB)
----.github()
--------workflows()
----README.md(1KB)
----transformer_in_transformer()
--------__init__.py(47B)
--------tnt.py(5KB)
----tnt.png(71KB)