fix(scripts): log node name differences in onnx models
This commit is contained in:
parent
7e730452f3
commit
f6db421631
|
@ -62,6 +62,18 @@ def diff_models(ref_model: ModelProto, cmp_model: ModelProto):
|
||||||
|
|
||||||
if diffs > 0:
|
if diffs > 0:
|
||||||
logger.warning("models have %s differences", diffs)
|
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:
|
else:
|
||||||
logger.info("models have no detectable differences")
|
logger.info("models have no detectable differences")
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue