search

ThanatosShinji / onnx-tool

A parser, editor and profiler tool for ONNX models.
https://pypi.org/project/onnx-tool/
MIT License
399 stars 52 forks source link

`model_profile` provided with extra argument in __main__.py #33

Open haojin2 opened 1 year ago

haojin2 commented 1 year ago

Describe the bug https://github.com/ThanatosShinji/onnx-tool/blob/main/onnx_tool/__main__.py#L93

To Reproduce Just run a model profile with the main.py like: python3 -m onnx_tool -m profile -i model.onnx

ThanatosShinji commented 1 year ago

Do you mean need more args for model profile using from console?

haojin2 commented 1 year ago

No, you need to remove , dump_outputs=args.names to make it work, otherwise you're providing an extra argument into that function call