huggingface / candle

Minimalist ML framework for Rust
Apache License 2.0
13.79k stars 751 forks source link

error when import onnx of yolo8 #2126

Open xuexl opened 2 weeks ago

xuexl commented 2 weeks ago

I train one model by yolo8, and export to a onnx file, then import by candel-onnx . An error accur:

Error: unsupported op_type Split for op NodeProto { input: ["/model.2/cv1/act/Mul_output_0", "onnx::Split_217"], output: ["/model.2/Split_output_0", "/model.2/Split_output_1"], name: "/model.2/Split", op_type: "Split", domain: "", attribute: [AttributeProto { name: "axis", ref_attr_name: "", doc_string: "", r#type: Int, f: 0.0, i: 1, s: [], t: None, g: None, sparse_tensor: None, tp: None, floats: [], ints: [], strings: [], tensors: [], graphs: [], sparse_tensors: [], type_protos: [] }], doc_string: "" }

Anybody can help me, please ? thanks .