Skip to content

Commit c539368

Browse files
committed
Merge remote-tracking branch 'upstream/master' into jignparm/optimize_transpose_multiply
2 parents 0fb855d + 14af496 commit c539368

File tree

4 files changed

+564
-3
lines changed

4 files changed

+564
-3
lines changed

tf2onnx/constants.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,5 +41,5 @@
4141
# Mapping opset to IR version.
4242
# When adding here, make sure that the IR changes don't impact that we do.
4343
OPSET_TO_IR_VERSION = {
44-
1: 3, 2:3, 3: 3, 4: 4, 5: 3, 6:3, 7:3, 8:4, 9:4, 10:5, 11:6, 12:7
44+
1: 3, 2: 3, 3: 3, 4: 4, 5: 3, 6: 3, 7: 3, 8: 4, 9: 4, 10: 5, 11: 6, 12: 7
4545
}

tf2onnx/graph.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1042,7 +1042,7 @@ def make_model(self, graph_doc, optimize=False, graph_name="tf2onnx", **kwargs):
10421042
# set the IR version based on opset
10431043
try:
10441044
model_proto.ir_version = constants.OPSET_TO_IR_VERSION.get(self.opset, model_proto.ir_version)
1045-
except:
1045+
except: # pylint: disable=bare-except
10461046
logger.error("ir_version override failed - install the latest onnx version")
10471047

10481048
# optimize the model proto.

tutorials/ConvertingSSDMobilenetToONNX.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -503,4 +503,4 @@
503503
},
504504
"nbformat": 4,
505505
"nbformat_minor": 2
506-
}
506+
}

0 commit comments

Comments
 (0)