Dilshod Tadjibaev
Dilshod Tadjibaev
I'll be working on a test harness to test various large models quickly.
RTMW3D-x is buildable with https://github.com/tracel-ai/burn/pull/3564 (still under review)
> i am able to import https://huggingface.co/Xenova/albert-large-v2/resolve/main/onnx/model.onnx but rust code has type errors I have 3 outstanding ONNX related PRs (#3563, #3564, #3550) with fixes. Most likely it's caused by...
Resnet is buildable:
Even though Yolo11X ONNX file can be converted into rust code, currently the generated rust code can't be built to due to not handing broadcasting (see why automatic full broadcasting...
I have added a harness to test models: https://github.com/tracel-ai/burn/tree/main/crates/burn-import/model-checks YOLOx11 is passing for tch and ndarray backends but currently it's failing on metal due to: https://github.com/tracel-ai/burn/issues/3600 bug in metal. I...
CLIP ViT-B/32 text model ONNX: 1. Convertable to rust 2. Buildable 3. Runnable 4. Accurate compared with ONNX Runtime outputs The PR is under the review: https://github.com/tracel-ai/burn/pull/3623
CLIP ViT-B-32 fixes are merged. tch, metal and ndarray backends work. To test: ``` cd crates/burn-import/model-checks/clip-vit-b-32-text ./get_model.py #or uv run ./get_model.py or python ./get_model.py cargo run --release # tch default...
> I can confirm Retinaface works fine now too > > Arcface can be converted, but panics at runtime with the metal backend, probably some of the same issues as...
> [clip-ViT-B-32 Vision](https://huggingface.co/Qdrant/clip-ViT-B-32-vision/blob/main/model.onnx) can be converted, but the converted code fails to compile due to incompatible tensor dims > > ``` > error[E0308]: mismatched types > --> clip_vision.rs:835:42 > |...