Skip to content

Commit 7798e9b

Browse files
committed
remove useless make_model
1 parent e59757b commit 7798e9b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/run_pretrained_models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -251,8 +251,8 @@ def run_test(self, name, backend="caffe2", onnx_file=None, opset=None, extra_ops
251251
# convert model to onnx
252252
onnx_graph = self.to_onnx(sess.graph, opset=opset, extra_opset=extra_opset,
253253
shape_override=shape_override, input_names=inputs.keys())
254+
onnx_graph = optimizer.optimize_graph(onnx_graph)
254255
model_proto = onnx_graph.make_model("converted from tf2onnx")
255-
model_proto = optimizer.optimize_graph(onnx_graph).make_model("optimized")
256256
logger.info("To_ONNX, OK")
257257
if onnx_file:
258258
self.create_onnx_file(name, model_proto, inputs, onnx_file)

0 commit comments

Comments
 (0)