timm documentation

Gêmeos

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

Gêmeos

Categoria: ViT · Código: timm/models/twins.py · Arquiteturas: 6 · Pesos pré-treinados: 6 (6 com resultados no ImageNet-1k)

Twins-PCPVT (Pyramid Vision Transformer com codificações de posição condicionais) e Twins-SVT (autoatenção espacialmente separável, atenção localmente agrupada alternando com atenção global subsampleada).

Artigos

Uso

Veja o Início rápido para carregar pesos pré-treinados, pré-processar imagens e fazer fine-tuning, e Extração de recursos para usar os modelos como backbones.

>>> import timm
>>> model = timm.create_model('twins_pcpvt_small.in1k', pretrained=True)
>>> data_config = timm.data.resolve_model_data_config(model)

Suporte à API de modelos

RecursoTwins
mapas de recursos de features_only=True✓ (passos 4, 8, 16, 32)
forward_intermediates()✓
prune_intermediate_layers()✓
Qualquer tamanho de entrada, sem necessidade de alterações✓
set_input_size()–
dynamic_img_size=True–
set_grad_checkpointing()✗
Decaimento de LR por camada (group_matcher())✓
reset_classifier()✓

✓ suportado, ✓ (quadrado) funciona apenas para tamanhos quadrados, ✗ não suportado, – não necessário (o modelo já aceita qualquer tamanho de entrada). Verificado executando cada recurso em um modelo inicializado aleatoriamente (Twins com twins_svt_small em 224); espera-se que outras arquiteturas da mesma classe se comportem da mesma forma. Consulte a página Model API para mais detalhes.

Arquiteturas

ArquiteturaParâmetros (M)GMACsTamanho da entradaPesos pré-treinados
twins_pcpvt_base43.836.682241
twins_pcpvt_large60.999.842241
twins_pcpvt_small24.113.832241
twins_svt_base56.078.592241
twins_svt_large99.2715.152241
twins_svt_small24.062.942241

Os GMACs e as contagens de parâmetros são para o tamanho de entrada e o classificador padrão de cada arquitetura, dos resultados do benchmark.

Pesos pré-treinados

Top-1 é a acurácia de validação no ImageNet-1k de results-imagenet.csv no tamanho de avaliação listado, - se o peso não tiver resultado no ImageNet-1k (por exemplo, pesos de pré-treinamento ou de imagem-texto). Classes é o tamanho do classificador pré-treinado, nenhum para pesos sem um.

PesoTamanho da entradaTop-1 (%)Tamanho da avaliaçãoClassesLicençaHub
twins_pcpvt_base.in1k22482.7242241000apache-2.0timm
twins_pcpvt_large.in1k22483.1582241000apache-2.0timm
twins_pcpvt_small.in1k22481.1162241000apache-2.0timm
twins_svt_base.in1k22483.1222241000apache-2.0timm
twins_svt_large.in1k22483.7182241000apache-2.0timm
twins_svt_small.in1k22481.6942241000apache-2.0timm
Update on GitHub