@@ -247,16 +247,11 @@ def run_test(self, name, backend="caffe2", onnx_file=None, opset=None, extra_ops
247
247
graph_def , input_names , outputs = tf_loader .from_checkpoint (model_path , input_names , outputs )
248
248
elif self .model_type in ["saved_model" ]:
249
249
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 )
253
251
except OSError :
254
252
model_path = dir_name
255
253
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 )
260
255
graph_def , input_names , outputs = res [:3 ]
261
256
elif self .model_type in ["keras" ]:
262
257
graph_def , input_names , outputs = tf_loader .from_keras (model_path , input_names , outputs )
0 commit comments