From 5cb28ed2cf730402f61fef120821cd9035c85854 Mon Sep 17 00:00:00 2001 From: Macrobull Date: Sun, 26 May 2019 14:52:28 +0800 Subject: [PATCH] add new ONNX polish_model --- onnx2fluid/onnx2fluid/onnx_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/onnx2fluid/onnx2fluid/onnx_utils.py b/onnx2fluid/onnx2fluid/onnx_utils.py index b6f6dc3..1c86b4b 100644 --- a/onnx2fluid/onnx2fluid/onnx_utils.py +++ b/onnx2fluid/onnx2fluid/onnx_utils.py @@ -642,7 +642,7 @@ if __name__ == '__main__': model = convert_version(model, 9) model = polish_model(model) - onnx.save(model, '/tmp/export.optimized.onnx') + onnx.save(model, '/tmp/export.polished.onnx') graph = model.graph value_info = inferred_model_value_info(model) -- GitLab