fabio-sim/LightGlue-ONNX

Can't run on CPU

XL634663985 opened this issue · 2 comments

File "D:\python-project\LightGlue-ONNX-main\test.py", line 17, in
print(runner.run(image0, image1, scales0, scales1))
File "D:\python-project\LightGlue-ONNX-main\onnx_runner\lightglue.py", line 53, in run
matches0, mscores0 = self.lightglue.run(
File "C:\Users\admin\AppData\Local\Programs\Python\Python38\lib\site-packages\onnxruntime\capi\onnxruntime_inference_collection.py", line 220, in run
return self._sess.run(output_names, input_feed, run_options)
onnxruntime.capi.onnxruntime_pybind11_state.RuntimeException: [ONNXRuntimeError] : 6 : RUNTIME_EXCEPTION : Non-zero status code returned while running MultiHeadAttention node. Name:'MultiHeadAttention_0' Status Message: Packed QKV of shape (B, L, N, 3, H) not implemented for CPU

Hi @XL634663985, thank you for your interest in LightGlue-ONNX.

I just added CPU support (#46). You can try out the models with *_fused_cpu.onnx at https://github.com/fabio-sim/LightGlue-ONNX/releases/tag/v1.0.0.

Thank you so much! This problem is solved!