Skip to content

Commit 5798722

Browse files
committed
Update run_pretrained_models.py
1 parent 82a87b1 commit 5798722

File tree

1 file changed

+2
-7
lines changed

1 file changed

+2
-7
lines changed

tests/run_pretrained_models.py

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -247,16 +247,11 @@ def run_test(self, name, backend="caffe2", onnx_file=None, opset=None, extra_ops
247247
graph_def, input_names, outputs = tf_loader.from_checkpoint(model_path, input_names, outputs)
248248
elif self.model_type in ["saved_model"]:
249249
try:
250-
res = tf_loader.from_saved_model(
251-
model_path, input_names, outputs, self.tag, self.signatures,
252-
self.concrete_function, self.large_model)
250+
res = tf_loader.from_saved_model(model_path, input_names, outputs, self.tag)
253251
except OSError:
254252
model_path = dir_name
255253
logger.info("Load model(2) from %r", model_path)
256-
res = tf_loader.from_saved_model(
257-
model_path, input_names, outputs, self.tag, self.signatures,
258-
self.concrete_function, self.large_model)
259-
# graph_def, input_names, outputs, concrete_func, imported = res
254+
res = tf_loader.from_saved_model(model_path, input_names, outputs, self.tag)
260255
graph_def, input_names, outputs = res[:3]
261256
elif self.model_type in ["keras"]:
262257
graph_def, input_names, outputs = tf_loader.from_keras(model_path, input_names, outputs)

0 commit comments

Comments
 (0)