timm documentation

EfficientFormer

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

EfficientFormer

Categoría: Efficient/Mobile · Código: timm/models/efficientformer.py · Arquitecturas: 3 · Pesos preentrenados: 3 (3 con resultados en ImageNet-1k)

EfficientFormer es un vision transformer diseñado para la latencia en dispositivos móviles, con bloques basados en pooling sobre mapas de características 4D en las primeras etapas y bloques de autoatención multi-cabeza en la última etapa.

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

Soporte de la API de modelos

CaracterísticaEfficientFormer
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 (EfficientFormer con efficientformer_l1 en 224); 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
efficientformer_l112.291.302241
efficientformer_l331.413.932241
efficientformer_l782.2310.172241

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.

PesoTamaño de entradaTop-1 (%)Tamaño de evaluaciónClasesLicenciaHub
efficientformer_l1.snap_dist_in1k22480.5022241000apache-2.0timm
efficientformer_l3.snap_dist_in1k22482.5622241000apache-2.0timm
efficientformer_l7.snap_dist_in1k22483.3962241000apache-2.0timm

Cita

@article{li2022efficientformer,
  title={EfficientFormer: Vision Transformers at MobileNet Speed},
  author={Li, Yanyu and Yuan, Geng and Wen, Yang and Hu, Eric and Evangelidis, Georgios and Tulyakov,
   Sergey and Wang, Yanzhi and Ren, Jian},
  journal={arXiv preprint arXiv:2206.01191},
  year={2022}
}
Update on GitHub