Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

导出onnx出现warning #16

Open
githubzjj1 opened this issue May 15, 2024 · 1 comment
Open

导出onnx出现warning #16

githubzjj1 opened this issue May 15, 2024 · 1 comment

Comments

@githubzjj1
Copy link

/usr/local/lib/python3.10/dist-packages/torch/onnx/utils.py:1636: UserWarning: The exported ONNX model failed ONNX shape inference.The model will not be executable by the ONNX Runtime.If this is unintended and you believe there is a bug,please report an issue at https://github.com/pytorch/pytorch/issues.Error reported by strict ONNX shape inference: [ShapeInferenceError] (op_type:Add, node name: /Add): A typestr: T, has unsupported type: tensor(bool) (Triggered internally at ../torch/csrc/jit/serialization/export.cpp:1407.)
请问这个warning重要吗,如果会影响模型效果的话应该怎么解决
这是使用ChatGLM3导出onnx的脚本时出现的问题

@githubzjj1
Copy link
Author

噢,我发现用我这个导出来的bmodel不会产生回复,那这个warning应该怎么解决呢

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant