timm documentation
TNT
TNT
Categoría: ViT · Código: timm/models/tnt.py · Arquitecturas: 3 · Pesos preentrenados: 3 (3 con resultados en ImageNet-1k)
Los modelos Transformer in Transformer (TNT) modelan la estructura dentro de cada parche mediante un transformador interno sobre subparches, junto con el transformador externo sobre parches.
Papers
Uso
Consulte la Introducción rápida para cargar pesos preentrenados, preprocesar imágenes y realizar el ajuste fino, y Extracción de características para usar los modelos como columnas verticales.
>>> import timm
>>> model = timm.create_model('tnt_s_patch16_224.in1k', pretrained=True)
>>> data_config = timm.data.resolve_model_data_config(model)Soporte de la API de modelos
✓ compatible, ✓ (cuadrado) funciona solo con tamaños cuadrados, ✗ no compatible, – no necesario (el modelo ya acepta cualquier tamaño de entrada). Verificado ejecutando cada función en un modelo inicializado aleatoriamente (TNT con tnt_s_legacy_patch16_224 a 224), se espera que otras arquitecturas de la misma clase se comporten de la misma manera. Consulte la página Model API para obtener más detalles.
Arquitecturas
| Arquitectura | Parámetros (M) | GMACs | Tamaño de entrada | Pesos preentrenados |
|---|---|---|---|---|
tnt_b_patch16_224 | 65.43 | 14.10 | 224 | 1 |
tnt_s_legacy_patch16_224 | 23.76 | 5.24 | 224 | 1 |
tnt_s_patch16_224 | 23.77 | 5.25 | 224 | 1 |
Los GMAC y el número de parámetros corresponden al tamaño de entrada y al clasificador predeterminados de cada arquitectura, según los resultados de las pruebas de rendimiento.
Pesos preentrenados
Top-1 es la precisión de validación en ImageNet-1k desde results-imagenet.csv en el tamaño de evaluación indicado, - si el peso no tiene un resultado de ImageNet-1k (p. ej. pesos de preentrenamiento o de imagen-texto). Classes es el tamaño del clasificador preentrenado, ninguno para pesos sin uno.
| Peso | Tamaño de entrada | Top-1 (%) | Tamaño de evaluación | Clases | Licencia | Hub |
|---|---|---|---|---|---|---|
tnt_b_patch16_224.in1k | 224 | 82.864 | 224 | 1000 | apache-2.0 | timm |
tnt_s_legacy_patch16_224.in1k | 224 | 81.512 | 224 | 1000 | apache-2.0 | timm |
tnt_s_patch16_224.in1k | 224 | 81.506 | 224 | 1000 | apache-2.0 | timm |