文件名称：变压器中的变压器：Pytorch中的Transformer在Transformer中的实现，像素级别注意与补丁级别注意配对以进行图像分类
文件大小：76KB
文件格式：ZIP
更新时间：2024-03-23 05:34:38
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)