You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
The text was updated successfully, but these errors were encountered:
robvoid
changed the title
[pnnx] onnx的unsqueeze参数是axes支持多个维度,torch的只能单个维度,pnnx转出来的ncnn和pnnx都有一些问题
[pnnx] onnx的unsqueeze参数支持多个维度,torch的只能单个维度,pnnx转换onnx模型存在问题
Aug 30, 2024
error log | 日志或报错信息 | ログ
运行 python xxx_pnnx.py错误:
TypeError: unsqueeze(): argument 'dim' must be int, not tuple
运行 python xxx_ncnn.py不报错,但是输出tensor的shape错误
model | 模型 | モデル
unsqueeze_error.sim.onnx.zip
onnxsim unsqueeze_error.onnx unsqueeze_error.sim.onnx
how to reproduce | 复现步骤 | 再現方法
预期输出shape [1, 7, 1, 1],ncnn输出shape [1, 1, 7]
The text was updated successfully, but these errors were encountered: