Skip to content

Commit 043475e

Browse files
committed
set continue_on_error to false for run pretrained model
1 parent 7b78248 commit 043475e

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

tests/run_pretrained_models.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -153,7 +153,7 @@ def run_tensorflow(self, sess, inputs):
153153

154154
def to_onnx(self, tf_graph, opset=None, shape_override=None, input_names=None):
155155
"""Convert graph to tensorflow."""
156-
return process_tf_graph(tf_graph, continue_on_error=True, verbose=True, opset=opset,
156+
return process_tf_graph(tf_graph, continue_on_error=False, verbose=True, opset=opset,
157157
target=Test.target, shape_override=shape_override,
158158
input_names=input_names, output_names=self.output_names)
159159

@@ -186,7 +186,6 @@ def run_onnxruntime(self, name, model_proto, inputs):
186186
"""Run test against msrt-next backend."""
187187
import onnxruntime as rt
188188
model_path = utils.save_onnx_model(TEMP_DIR, name, inputs, model_proto, include_test_data=True)
189-
utils.save_onnx_model(TEMP_DIR, name, inputs, model_proto, include_test_data=False, as_text=True)
190189
print("\t\t" + model_path)
191190
m = rt.InferenceSession(model_path)
192191
results = m.run(self.output_names, inputs)

0 commit comments

Comments
 (0)