Skip to content

Commit e96320a

Browse files
committed
fix --saved_model option
1 parent b420b28 commit e96320a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tf2onnx/convert.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -107,7 +107,7 @@ def main():
107107
model_path = args.saved_model
108108

109109
# todo: consider to enable const folding by default?
110-
graph_def = tf_optimize(args.inputs, args.outputs, graph_def, args.fold_const)
110+
graph_def = tf_optimize(inputs, outputs, graph_def, args.fold_const)
111111

112112
with tf.Graph().as_default() as tf_graph:
113113
tf.import_graph_def(graph_def, name='')

0 commit comments

Comments
 (0)