fix(scripts): log node name differences in onnx models

ssube · Sep 5, 2023 · f6db421 · f6db421
1 parent 7e73045
commit f6db421
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/api/scripts/onnx-diff.py b/api/scripts/onnx-diff.py
@@ -62,6 +62,18 @@ def diff_models(ref_model: ModelProto, cmp_model: ModelProto):
 
   if diffs > 0:
     logger.warning("models have %s differences", diffs)
+
+    ref_init_names = set([init.name for init in ref_model.graph.initializer])
+    cmp_init_names = set([init.name for init in cmp_model.graph.initializer])
+    ref_node_names = set([node.name for node in ref_model.graph.node])
+    cmp_node_names = set([node.name for node in cmp_model.graph.node])
+
+    logger.info("different names: %s, %s, %s, %s",
+                ref_init_names.difference(cmp_init_names),
+                cmp_init_names.difference(ref_init_names),
+                ref_node_names.difference(cmp_node_names),
+                cmp_node_names.difference(ref_node_names)
+                )
   else:
     logger.info("models have no detectable differences")