timm documentation

ConViT

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

ConViT

Kategori: ViT · Kod: timm/models/convit.py · Mimariler: 3 · Önceden eğitilmiş ağırlıklar: 3 (3'ü ImageNet-1k sonuçlarıyla)

ConViT, erken Vision Transformer bloklarındaki öz-dikkati, yumuşak konvolüsyonel indüktif önyargı olarak konvolüsyon gibi davranacak şekilde başlatılabilecek kapılı konumsal öz-dikkat (GPSA) ile değiştirir.

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

Model API desteği

ÖzellikConVit
features_only=True özellik haritaları✓ (adım 16, 16, 16)
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 (ConVit, 224 boyutunda convit_tiny ile); aynı sınıftaki diğer mimarilerin 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
convit_base86.5417.522241
convit_small27.785.762241
convit_tiny5.711.262241

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
convit_base.fb_in1k22482.2962241000apache-2.0timm
convit_small.fb_in1k22481.4122241000apache-2.0timm
convit_tiny.fb_in1k22473.1402241000apache-2.0timm

Atıf

@article{d2021convit,
  title={ConViT: Improving Vision Transformers with Soft Convolutional Inductive Biases},
  author={d'Ascoli, St{'e}phane and Touvron, Hugo and Leavitt, Matthew and Morcos, Ari and Biroli, Giulio and Sagun, Levent},
  journal={arXiv preprint arXiv:2103.10697},
  year={2021}
}
Update on GitHub