提交 5cb28ed2 编写于 作者: M Macrobull

add new ONNX polish_model

上级 492e9661
...@@ -642,7 +642,7 @@ if __name__ == '__main__': ...@@ -642,7 +642,7 @@ if __name__ == '__main__':
model = convert_version(model, 9) model = convert_version(model, 9)
model = polish_model(model) model = polish_model(model)
onnx.save(model, '/tmp/export.optimized.onnx') onnx.save(model, '/tmp/export.polished.onnx')
graph = model.graph graph = model.graph
value_info = inferred_model_value_info(model) value_info = inferred_model_value_info(model)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册