timm documentation

TinyViT

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

TinyViT

Kategori: Hibrit · Kod: timm/models/tiny_vit.py · Mimariler: 5 · Önceden eğitilmiş ağırlıklar: 11 (8'i ImageNet-1k sonuçlarıyla)

TinyViT, büyük öğretmen modellerinden hızlı damıtma ile ön eğitilmiş küçük hiyerarşik görme dönüştürücüleri ailesidir (ilk aşama konvolüsyonel, ardından yerel pencere dikkati).

Makaleler

Kullanım

Önceden eğitilmiş ağırlıkları yükleme, görüntü ön işleme ve ince ayar için Hızlı Başlangıç bölümüne, modelleri omurga olarak kullanmak için ise Özellik Çıkarımı bölümüne bakın.

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

Model API desteği

ÖzellikTinyVit
features_only=True özellik haritaları✓ (adım 4, 8, 16, 32)
forward_intermediates()✓
prune_intermediate_layers()✓
Herhangi bir giriş boyutu, değişiklik gerekmez✓
set_input_size()–
dynamic_img_size=True–
set_grad_checkpointing()✓
Katman bazlı LR azaltma (group_matcher())✓
reset_classifier()✓

✓ destekleniyor, ✓ (kare) yalnızca kare boyutlarda çalışır, ✗ desteklenmiyor, – gerekli değil (model zaten herhangi bir giriş boyutunu kabul eder). Her özellik, rastgele başlatılmış bir model üzerinde çalıştırılarak kontrol edilir (TinyVit, 224 boyutunda tiny_vit_5m_224), aynı sınıfın diğer mimarilerinin de aynı şekilde davranması beklenir. Ayrıntılar için Model API sayfasına bakın.

Mimariler

MimariParametreler (M)GMAC'larGirdi boyutuÖnceden eğitilmiş ağırlıklar
tiny_vit_5m_22412.081.272243
tiny_vit_11m_22420.352.032243
tiny_vit_21m_22433.214.272243
tiny_vit_21m_38421.2213.723841
tiny_vit_21m_51221.2626.935121

GMAC'lar ve parametre sayıları, her mimarinin varsayılan giriş boyutu ve sınıflandırıcısı içindir ve benchmark sonuçlarından alınmıştır.

Önceden eğitilmiş ağırlıklar

Top-1, belirtilen değerlendirme boyutunda results-imagenet.csv dosyasındaki ImageNet-1k doğrulama doğruluğudur; ağırlığın ImageNet-1k sonucu yoksa (ör. ön eğitim veya görsel-metin ağırlıkları) bu değer "-" olarak gösterilir. Classes, ön eğitilmiş sınıflandırıcının boyutudur; sınıflandırıcısı olmayan ağırlıklar için bu değer "none"dur.

AğırlıkGirdi boyutuEn iyi 1 (%)Değerlendirme boyutuSınıflarLisansHub
tiny_vit_5m_224.dist_in22k224--21841apache-2.0timm
tiny_vit_5m_224.dist_in22k_ft_in1k22480.8742241000apache-2.0timm
tiny_vit_5m_224.in1k22479.1962241000apache-2.0timm
tiny_vit_11m_224.dist_in22k224--21841apache-2.0timm
tiny_vit_11m_224.dist_in22k_ft_in1k22483.2322241000apache-2.0timm
tiny_vit_11m_224.in1k22481.5382241000apache-2.0timm
tiny_vit_21m_224.dist_in22k224--21841apache-2.0timm
tiny_vit_21m_224.dist_in22k_ft_in1k22485.0882241000apache-2.0timm
tiny_vit_21m_224.in1k22483.2682241000apache-2.0timm
tiny_vit_21m_384.dist_in22k_ft_in1k38486.0883841000apache-2.0timm
tiny_vit_21m_512.dist_in22k_ft_in1k51286.4445121000apache-2.0timm
Update on GitHub