用torch导出的模型有时候参数过多,不利于查看。查找资料onnxsim可以简化模型,让显示更加自然。
如,reshap层的导出,红框中细节参数被显示出现。默认导出:
pip install onnx-simplifier
ERROR: No matching distribution found for onnxsim
阅读 121
2023-01-20
用torch导出的模型有时候参数过多,不利于查看。查找资料onnxsim可以简化模型,让显示更加自然。
如,reshap层的导出,红框中细节参数被显示出现。默认导出:
pip install onnx-simplifier
相关推荐
精彩评论(0)