song@song-Mi-Gaming-Laptop-15-6:~/onnx2trt$ python -m onnxsim cls.onnx cls1.onnxSimplifying...
Traceback (most recent call last):
File "/home/song/anaconda3/lib/python3.7/runpy.py", line 193, in _run_module_as_main
"__main__", mod_spec)
File "/home/song/anaconda3/lib/python3.7/runpy.py", line 85, in _run_code
exec(code, run_globals)
File "/home/song/anaconda3/lib/python3.7/site-packages/onnxsim/__main__.py", line 90, in <module>
main()
File "/home/song/anaconda3/lib/python3.7/site-packages/onnxsim/__main__.py", line 78, in main
custom_lib=args.custom_lib)
File "/home/song/anaconda3/lib/python3.7/site-packages/onnxsim/onnx_simplifier.py", line 492, in simplify
model = fixed_point(model, infer_shapes_and_optimize, constant_folding)
File "/home/song/anaconda3/lib/python3.7/site-packages/onnxsim/onnx_simplifier.py", line 393, in fixed_point
x = func_b(x)
File "/home/song/anaconda3/lib/python3.7/site-packages/onnxsim/onnx_simplifier.py", line 486, in constant_folding
custom_lib=custom_lib)
File "/home/song/anaconda3/lib/python3.7/site-packages/onnxsim/onnx_simplifier.py", line 232, in forward_for_node_outputs
custom_lib=custom_lib)
File "/home/song/anaconda3/lib/python3.7/site-packages/onnxsim/onnx_simplifier.py", line 210, in forward
inputs.update(generate_specific_rand_input(model, {name: shape}))
File "/home/song/anaconda3/lib/python3.7/site-packages/onnxsim/onnx_simplifier.py", line 102, in generate_specific_rand_input
'Run "python3 -m onnxsim -h" for details'.format(key, shape))
RuntimeError: The shape of input "x" has dynamic size "[-1, 3, 640, 640]", please determine the input size manually by "--dynamic-input-shape --input-shape xxx" or "--input-shape xxx". Run "python3 -m onnxsim -h" for details
......
最新推荐文章于 2023-04-18 14:05:18 发布