daquexian / onnx-simplifier

Simplify your onnx model

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

[咨询] 请问下onnxsim支持多输入 dynamic shape吗

wjj19950828 opened this issue · comments

Describe the bug
如果是多输入的话,动态shape应该如何使用呢

单输入:

python -m onnxsim --dynamic-input-shape --input-shape="input_ids:1,10" model.onnx model_sim.onnx 

多输入尝试了好几种都不太行

感谢~

你好,可以在网页版 https://convertmodel.com/#input=onnx&output=onnx 上试一下吗

@daquexian 您好,好像都会报错

没有命令行可以支持吗

😂 暂时还没有,还在开发中,可能这两天发布吧

同求

请问开发好了吗?我也遇到了这个问题,部分动态输入,部分又是固定输入

using the format below:

python3 -m onnxsim original.onnx dest_sim.onnx --input-shape input1_name:num1,num2,num3 input2_name:num4,num5

Update: The support of dynamic input shape has been greatly improved since v0.4.0. --dynamic-input-shape and --input-shape arguments are NOT NEEDED anymore. For example:

Old ❌ :

onnxsim input.onnx output.onnx --dynamic-input-shape --input-shape 1,3,224,224

New ✔️:

onnxsim input.onnx output.onnx

Please install the latest version and have a try!