Open 7kanak opened 3 months ago
Looking at the panicking statement in question, seems to be an issue with the unsqueeze
op.
Right now most of the ONNX ops are parsed during import expecting some of the shapes to be available in the protobuf metadata, but in practice that is not always the case (as seen here).
I am also having same error @laggui
In previous versions the unsqueeze onnx op was also checking the input shapes but it wasn't required. This has been changed on main if you want to try it out. It should not give you this error 🙂
I was following the onnx import example in burn
examples/onnx-inference
. but instead of using mnist.py to generate the onnx model, i am using transformers python package to generate onnx modelgetting this error
you can find the onnx model file at https://github.com/7kanak/rust_ml/blob/8ac45531fcb6b789ed685890c94011856671cd80/bert-ft/src/model/albert.onnx