mosure/burn_dino

burn dinov2 model inference and training 🔥🦖

burn_dino 🔥🦖

burn dinov2 model, view the live demo

input	fg pca

cargo run --example pca cargo run -p bevy_burn_dino

features

• inference
• pca transform layer
• ViT configurations
• real-time camera demo
• training (loss + dropout)
• optimized attention
• automatic weights cache/download
• quantization
• feature upsampling

setup

• download pre-trained model (ViT-S|B, /wo registers) from here
• place in ./assets/models
• run import tool cargo run --bin import

benchmarks

• cargo bench
• open target/criterion/report/index.html