文件名称：transformer_in_transformer_flax
文件大小：7KB
文件格式：ZIP
更新时间：2024-04-01 10:34:10
Python
JAX / Flax中的Transformer中的Transformer 此存储库实现 ，像素级注意与补丁级注意配对以进行图像分类。 Pytorch实施和Vision Transformer回购在很大程度上受到了启发。 安装 $ pip install transformer-in-transformer-flax 用法 from jax import random from jax import numpy as jnp from transformer_in_transformer_flax import TransformerInTransformer , TNTConfig #example configuration for TNT-B config = TNTConfig ( num_classes = 1000 , depth = 12 , imag
【文件预览】：
transformer_in_transformer_flax-master
----.github()
--------workflows()
----.idea()
--------transformer_in_transformer_flax.iml(553B)
--------.gitignore(47B)
----LICENSE(1KB)
----setup.py(833B)
----README.md(2KB)
----transformer_in_transformer_flax()
--------__init__.py(84B)
--------tnt.py(7KB)
----.gitignore(54B)