apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

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

This commit is contained in:
Sean Sube 2023-09-04 21:42:06 -05:00
parent 7e730452f3
commit f6db421631
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
api/scripts/onnx-diff.py
View File

@ -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")