Assertion failed: inputs.at(2).is_weights

x33g5p2x  于2022-07-22 转载在 其他  
字(0.5k)|赞(0)|评价(0)|浏览(332)

Assertion failed: inputs.at(2).is_weights

安装onnxsim,两种方式:

1.  pip install onnxsim,

  1. pip install onnx-simplifier

加载onnx文件,simplify处理后重新保存,代码如下:

import onnx
from onnxsim import simplify
onnx_model = onnx.load(output_path)  # load onnx model
model_simp, check = simplify(onnx_model)
assert check, "Simplified ONNX model could not be validated"
onnx.save(model_simp, output_path)
print('finished exporting onnx')

————————————————
版权声明:本文为CSDN博主「人类高质量算法工程师」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/qq_35037684/article/details/121248694

相关文章