timm documentation

DeiT

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

DeiT

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

Veri verimli Görüntü Dönüşümcüleri (DeiT), yalnızca ImageNet-1k üzerinde eğitilen, isteğe bağlı olarak bir CNN öğretmeninden öğrenen bir damıtma jetonu içeren Görüntü Dönüşümcüleridir. DeiT III modelleri, geliştirilmiş bir tarif ile eğitilir.

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

Model API desteği

ÖzellikVisionTransformerVisionTransformerDistilled
features_only=True özellik haritaları✓ (adım 16, 16, 16)✓ (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 (VisionTransformer 224 için deit_tiny_patch16_224, VisionTransformerDistilled 224 için deit_tiny_distilled_patch16_224), aynı sınıftaki diğer mimarilerin aynı şekilde davranması beklenir. Ayrıntılar için Model API sayfasına bakın.

Mimariler

MimariSınıfParametreler (M)GMAC'larGirdi boyutuÖnceden eğitilmiş ağırlıklar
deit3_base_patch16_224VisionTransformer86.5917.582242
deit3_base_patch16_384VisionTransformer86.8855.543842
deit3_huge_patch14_224VisionTransformer632.13167.402242
deit3_large_patch16_224VisionTransformer304.3761.602242
deit3_large_patch16_384VisionTransformer304.76191.213842
deit3_medium_patch16_224VisionTransformer38.858.002242
deit3_small_patch16_224VisionTransformer22.064.612242
deit3_small_patch16_384VisionTransformer22.2115.523842
deit_base_distilled_patch16_224VisionTransformerDistilled87.3417.682241
deit_base_distilled_patch16_384VisionTransformerDistilled87.6355.653841
deit_base_patch16_224VisionTransformer86.5717.582241
deit_base_patch16_384VisionTransformer86.8655.543841
deit_small_distilled_patch16_224VisionTransformerDistilled22.444.632241
deit_small_patch16_224VisionTransformer22.054.612241
deit_tiny_distilled_patch16_224VisionTransformerDistilled5.911.272241
deit_tiny_patch16_224VisionTransformer5.721.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
deit3_base_patch16_224.fb_in1k22483.8002241000apache-2.0timm
deit3_base_patch16_224.fb_in22k_ft_in1k22485.7282241000apache-2.0timm
deit3_base_patch16_384.fb_in1k38485.0883841000apache-2.0timm
deit3_base_patch16_384.fb_in22k_ft_in1k38486.7423841000apache-2.0timm
deit3_huge_patch14_224.fb_in1k22485.2042241000apache-2.0timm
deit3_huge_patch14_224.fb_in22k_ft_in1k22487.1802241000apache-2.0timm
deit3_large_patch16_224.fb_in1k22484.7722241000apache-2.0timm
deit3_large_patch16_224.fb_in22k_ft_in1k22486.9902241000apache-2.0timm
deit3_large_patch16_384.fb_in1k38485.8203841000apache-2.0timm
deit3_large_patch16_384.fb_in22k_ft_in1k38487.7363841000apache-2.0timm
deit3_medium_patch16_224.fb_in1k22483.0902241000apache-2.0timm
deit3_medium_patch16_224.fb_in22k_ft_in1k22484.5502241000apache-2.0timm
deit3_small_patch16_224.fb_in1k22481.3822241000apache-2.0timm
deit3_small_patch16_224.fb_in22k_ft_in1k22483.0822241000apache-2.0timm
deit3_small_patch16_384.fb_in1k38483.4323841000apache-2.0timm
deit3_small_patch16_384.fb_in22k_ft_in1k38484.8463841000apache-2.0timm
deit_base_distilled_patch16_224.fb_in1k22483.3842241000apache-2.0timm
deit_base_distilled_patch16_384.fb_in1k38485.4383841000apache-2.0timm
deit_base_patch16_224.fb_in1k22481.9802241000apache-2.0timm
deit_base_patch16_384.fb_in1k38483.1083841000apache-2.0timm
deit_small_distilled_patch16_224.fb_in1k22481.2042241000apache-2.0timm
deit_small_patch16_224.fb_in1k22479.8562241000apache-2.0timm
deit_tiny_distilled_patch16_224.fb_in1k22474.5382241000apache-2.0timm
deit_tiny_patch16_224.fb_in1k22472.1902241000apache-2.0timm
Update on GitHub