Skip to content

Commit 5c5e54a

Browse files
authored
Merge pull request #320 from csteegz/patch-1
Fall back to non optimized model proto
2 parents 607d780 + 049ddf4 commit 5c5e54a

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tf2onnx/convert.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -102,6 +102,8 @@ def main():
102102
output_names=args.outputs,
103103
inputs_as_nchw=args.inputs_as_nchw)
104104

105+
model_proto = g.make_model("converted from {}".format(args.input))
106+
105107
new_model_proto = GraphUtil.opt_transposes_with_graph(g, "converted from {}".format(args.input),
106108
optimize=not args.continue_on_error)
107109
if new_model_proto:

0 commit comments

Comments
 (0)