timm documentation

EfficientViM

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

EfficientViM

Kategori: Efficient/Mobile Β· Kode: timm/models/efficientvim.py Β· Arsitektur: 8 Β· Bobot terlatih: 12 (0 dengan hasil ImageNet-1k)

Efficient Vision Mamba (EfficientViM) dibangun di sekitar lapisan Hidden State Mixer based State Space Duality (HSM-SSD), yang diimplementasikan di sini dalam PyTorch murni. Model _dist menambahkan kepala distilasi.

Makalah

Penggunaan

Lihat Quickstart untuk memuat bobot terlatih, prapemrosesan gambar, dan penyesuaian halus, serta Ekstraksi Fitur untuk menggunakan model sebagai tulang punggung.

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

Dukungan API Model

FiturEfficientViMEfficientViMDistilled
feature maps features_only=Trueβœ“ (strides 16, 32, 64)βœ“ (strides 16, 32, 64)
forward_intermediates()βœ“βœ“
prune_intermediate_layers()βœ“βœ“
Ukuran input apa pun, tidak perlu perubahanβœ“βœ“
set_input_size()––
dynamic_img_size=True––
set_grad_checkpointing()βœ“βœ“
Penurunan LR per lapisan (group_matcher())βœ“βœ“
reset_classifier()βœ“βœ“

βœ“ didukung, βœ“ (persegi) hanya berfungsi untuk ukuran persegi, βœ— tidak didukung, – tidak diperlukan (model sudah menerima ukuran input apa pun). Diperiksa dengan menjalankan setiap fitur pada model yang diinisialisasi secara acak (EfficientViM dengan efficientvim_m1 pada 224, EfficientViMDistilled dengan efficientvim_m1_dist pada 224), arsitektur lain dari kelas yang sama diharapkan berperilaku sama. Lihat halaman Model API untuk detailnya.

Arsitektur

ArsitekturKelasParameter (M)GMACsUkuran inputBobot terlatih
efficientvim_m1EfficientViM6.68-2242
efficientvim_m1_distEfficientViMDistilled7.64-2241
efficientvim_m2EfficientViM13.90-2242
efficientvim_m2_distEfficientViMDistilled15.32-2241
efficientvim_m3EfficientViM16.60-2242
efficientvim_m3_distEfficientViMDistilled18.18-2241
efficientvim_m4EfficientViM19.61-2562
efficientvim_m4_distEfficientViMDistilled21.18-2561

GMACs dan jumlah parameter adalah untuk ukuran input dan klasifer default dari setiap arsitektur, dari hasil benchmark.

Bobot terlatih

Top-1 adalah akurasi validasi ImageNet-1k dari results-imagenet.csv pada ukuran eval yang tercantum, - jika bobot tidak memiliki hasil ImageNet-1k (misalnya bobot pretraining atau image-text). Classes adalah ukuran dari classifier pretrained, none untuk bobot yang tidak memilikinya.

BobotUkuran inputTop-1 (%)Ukuran evaluasiKelasLisensiHub
efficientvim_m1.e300_in1k224--1000mittimm
efficientvim_m1.e450_in1k224--1000mittimm
efficientvim_m1_dist.in1k224--1000mittimm
efficientvim_m2.e300_in1k224--1000mittimm
efficientvim_m2.e450_in1k224--1000mittimm
efficientvim_m2_dist.in1k224--1000mittimm
efficientvim_m3.e300_in1k224--1000mittimm
efficientvim_m3.e450_in1k224--1000mittimm
efficientvim_m3_dist.in1k224--1000mittimm
efficientvim_m4.e300_in1k256--1000mittimm
efficientvim_m4.e450_in1k256--1000mittimm
efficientvim_m4_dist.in1k256--1000mittimm
Update on GitHub