timm documentation

LeViT

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

LeViT

الفئة: هجينة · الكود: timm/models/levit.py · الهندسات: 20 · الأوزان المُسبقة التدريب: 10 (10 منها بنتائج على ImageNet-1k)

LeViT هو شبكة هجينة للاستدلال السريع، تتكون من جذر تجميعي (convolutional stem) يليه مراحل انتباه تقلص الدقة عبر أخذ عينات أدنى (downsampling) قائم على الانتباه، وتستخدم انحيازات الانتباه (attention biases) للموقع.

الأوراق البحثية

الاستخدام

راجع البدء السريع لتحميل الأوزان المُسبقة التدريب ومعالجة الصور مسبقًا وضبط الدقة، وراجع استخراج الميزات لاستخدام النماذج كعمود فقري.

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

دعم واجهة برمجة التطبيقات للنماذج

ميزةLevitLevitDistilled
خرائط الميزات features_only=True✓ (خطوات 16، 32، 64)✓ (خطوات 16، 32، 64)
forward_intermediates()✓✓
prune_intermediate_layers()✓✓
أي حجم إدخال، دون الحاجة إلى أي تغييرات✗✗
set_input_size()✗✗
dynamic_img_size=True✗✗
set_grad_checkpointing()✓✓
تلاشي معدل التعلم على مستوى الطبقات (group_matcher())✓✓
reset_classifier()✓✓

✓ مدعوم، ✓ (مربع) يعمل مع الأحجام المربعة فقط، ✗ غير مدعوم، – غير مطلوب (يقبل النموذج بالفعل أي حجم إدخال). تم التحقق من ذلك بتشغيل كل ميزة على نموذج مُهيأ عشوائياً (Levit مع levit_256d عند 224، وLevitDistilled مع levit_128s عند 224)، ومن المتوقع أن تتصرف البنى المعمارية الأخرى من نفس الفئة بنفس الطريقة. راجع صفحة Model API للتفاصيل.

البنى

البنيةفئةالمعاملات (مليون)GMACsحجم الإدخالالأوزان المُسبقة التدريب
levit_128LevitDistilled9.190.412241
levit_128sLevitDistilled7.760.302241
levit_192LevitDistilled10.920.662241
levit_256LevitDistilled18.861.132241
levit_256dLevit26.161.392240
levit_384LevitDistilled39.072.352241
levit_384_s8LevitDistilled39.069.952240
levit_512Levit95.085.622240
levit_512_s8Levit73.9721.772240
levit_512dLevit92.395.842240
levit_conv_128LevitDistilled9.190.412241
levit_conv_128sLevitDistilled7.760.302241
levit_conv_192LevitDistilled10.920.662241
levit_conv_256LevitDistilled18.861.132241
levit_conv_256dLevit26.161.392240
levit_conv_384LevitDistilled39.072.352241
levit_conv_384_s8LevitDistilled39.069.952240
levit_conv_512Levit95.085.622240
levit_conv_512_s8Levit73.9721.772240
levit_conv_512dLevit92.395.842240

تُحسب قيم GMACs وعدد المعلمات لحجم المدخلات الافتراضي والتصنيف الخاص بكل معمارية، وذلك بناءً على نتائج الاختبارات المرجعية.

الأوزان المُسبقة التدريب

Top-1 هو دقة التحقق على ImageNet-1k من results-imagenet.csv عند حجم التقييم المذكور، - إذا لم يكن للوزن نتيجة على ImageNet-1k (مثل أوزان ما قبل التدريب أو أوزان الصورة-النص). Classes هو حجم المصنِّف المُدرَّب مسبقًا، ولا يوجد شيء للأوزان التي لا تحتوي على مصنِّف.

الوزنحجم الإدخالأعلى 1 (%)حجم التقييمفئاترخصةالمركز
levit_128.fb_dist_in1k22478.4862241000apache-2.0timm
levit_128s.fb_dist_in1k22476.5202241000apache-2.0timm
levit_192.fb_dist_in1k22479.8582241000apache-2.0timm
levit_256.fb_dist_in1k22481.5062241000apache-2.0timm
levit_384.fb_dist_in1k22482.6002241000apache-2.0timm
levit_conv_128.fb_dist_in1k22478.4882241000apache-2.0timm
levit_conv_128s.fb_dist_in1k22476.5342241000apache-2.0timm
levit_conv_192.fb_dist_in1k22479.8602241000apache-2.0timm
levit_conv_256.fb_dist_in1k22481.5062241000apache-2.0timm
levit_conv_384.fb_dist_in1k22482.6002241000apache-2.0timm

الاستشهاد

@article{graham2021levit,
  title={LeViT: a Vision Transformer in ConvNet's Clothing for Faster Inference},
  author={Benjamin Graham and Alaaeldin El-Nouby and Hugo Touvron and Pierre Stock and Armand Joulin and Herv'e J'egou and Matthijs Douze},
  journal={arXiv preprint arXiv:2104.01136},
  year={2021}
}
Update on GitHub