Skip to content

Commit 7bafec5

Browse files
authored
better way to get kwargs (#1541)
Signed-off-by: Guenther Schmuelling <[email protected]>
1 parent daeab0d commit 7bafec5

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

tools/onnx-optimize.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -50,10 +50,7 @@ def main():
5050

5151
onnx_graph = g.make_graph(org_model_proto.graph.doc_string + " (+tf2onnx/onnx-optimize)")
5252

53-
kwargs = {"producer_name": org_model_proto.producer_name,
54-
"producer_version": org_model_proto.producer_version,
55-
"opset_imports": org_model_proto.opset_import,
56-
"ir_version": org_model_proto.ir_version}
53+
kwargs = GraphUtil.get_onnx_model_properties(org_model_proto)
5754

5855
model_proto = helper.make_model(onnx_graph, **kwargs)
5956

0 commit comments

Comments
 (0)