timm documentation

FastViT

You are viewing main version, which requires installation from source. If you'd like regular pip install, checkout the latest stable version (v1.0.31).
Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

FastViT

Categoría: Eficiente/Móvil · Código: timm/models/fastvit.py · Arquitecturas: 12 · Pesos preentrenados: 21 (14 con resultados en ImageNet-1k)

FastViT es una red híbrida convolucional / transformer que utiliza la reparametrización estructural para eliminar las conexiones de salto en la inferencia. El módulo también incluye los codificadores de imagen MobileCLIP (fastvit_mci*).

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('fastvit_sa24.apple_dist_in1k', pretrained=True)
>>> data_config = timm.data.resolve_model_data_config(model)

Soporte de la API de modelos

CaracterísticaFastVit
mapas de características de features_only=True✓ (pasos 4, 8, 16, 32)
forward_intermediates()✓
prune_intermediate_layers()✓
Cualquier tamaño de entrada, no se necesitan cambios✓
set_input_size()–
dynamic_img_size=True–
set_grad_checkpointing()✓
Decaimiento del LR por capa (group_matcher())✓
reset_classifier()✓

✓ 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 (FastVit con fastvit_t8 en 256); se espera que otras arquitecturas de la misma clase se comporten de la misma manera. Consulte la página Model API para más detalles.

Arquitecturas

ArquitecturaParámetros (M)GMACsTamaño de entradaPesos preentrenados
fastvit_ma3643.987.822562
fastvit_mci011.362.392562
fastvit_mci121.464.672561
fastvit_mci235.707.852562
fastvit_mci3125.0714.822561
fastvit_mci4321.5727.782561
fastvit_s129.431.802562
fastvit_sa1211.551.942562
fastvit_sa2421.503.772562
fastvit_sa3631.465.592562
fastvit_t84.000.692562
fastvit_t127.511.392562

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.

Los pesos se agrupan por origen: el artículo o procedencia indicado en la configuración preentrenada, en su defecto una organización del Hub distinta de timm, y si no, la primera parte de la etiqueta preentrenada (architecture.tag). Los orígenes con un solo peso se listan bajo other.

etiquetas de apple

PesoTamaño de entradaTop-1 (%)Tamaño de evaluaciónClasesLicenciaHub
fastvit_ma36.apple_dist_in1k25684.6082561000fastvit-licensetimm
fastvit_ma36.apple_in1k25683.8842561000fastvit-licensetimm
fastvit_mci0.apple_mclip2_dfndr2b256--512apple-amlrtimm
fastvit_mci2.apple_mclip2_dfndr2b256--512apple-amlrtimm
fastvit_mci3.apple_mclip2_dfndr2b256--768apple-amlrtimm
fastvit_mci4.apple_mclip2_dfndr2b256--768apple-amlrtimm
fastvit_s12.apple_dist_in1k25681.0742561000fastvit-licensetimm
fastvit_s12.apple_in1k25679.8822561000fastvit-licensetimm
fastvit_sa12.apple_dist_in1k25681.8382561000fastvit-licensetimm
fastvit_sa12.apple_in1k25680.8462561000fastvit-licensetimm
fastvit_sa24.apple_dist_in1k25683.3682561000fastvit-licensetimm
fastvit_sa24.apple_in1k25682.7062561000fastvit-licensetimm
fastvit_sa36.apple_dist_in1k25684.0282561000fastvit-licensetimm
fastvit_sa36.apple_in1k25683.5022561000fastvit-licensetimm
fastvit_t8.apple_dist_in1k25677.1702561000fastvit-licensetimm
fastvit_t8.apple_in1k25676.1842561000fastvit-licensetimm
fastvit_t12.apple_dist_in1k25680.3642561000fastvit-licensetimm
fastvit_t12.apple_in1k25679.2762561000fastvit-licensetimm

apple

Alojado por apple en el Hugging Face Hub.

PesoTamaño de entradaTop-1 (%)Tamaño de evaluaciónClasesLicenciaHub
fastvit_mci0.apple_mclip256--512apple-amlrapple/mobileclip_s0_timm
fastvit_mci1.apple_mclip256--512apple-amlrapple/mobileclip_s1_timm
fastvit_mci2.apple_mclip256--512apple-amlrapple/mobileclip_s2_timm
Update on GitHub