Skip to content

Latest commit

 

History

37 Commits

Folders and files

NameName
Last commit message
Last commit date
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

burn_dino 🔥🦖

test GitHub License GitHub Last Commit GitHub Releases GitHub Issues Average time to resolve an issue crates.io

burn dinov2 model, view the live demo

input fg pca
Alt text Alt text
Alt text Alt text
Alt text Alt text
Alt text Alt text

cargo run --features backend_wgpu --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 --features import --bin import

benchmarks

  • cargo bench --bench dinov2 --features backend_wgpu
  • open target/criterion/report/index.html

About

burn dino model inference and training 🔥🦖

Topics

Resources

Stars

6 stars

Watchers

1 watching

Forks

Used by

Contributors

Languages