apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(scripts): show data shape in initializer diff

This commit is contained in:
Sean Sube 2023-04-10 08:26:24 -05:00
parent 512f41135d
commit b4095cc2db
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -26,7 +26,9 @@ def diff_models(ref_model: ModelProto, cmp_model: ModelProto):
logger.info("different data types: %s vs %s", ref_init.data_type, cmp_init.data_type)
diffs += 1
elif len(ref_init.raw_data) != len(cmp_init.raw_data):
logger.info("different raw data size: %s vs %s", len(ref_init.raw_data), len(cmp_init.raw_data))
ref_data = to_array(ref_init)
cmp_data = to_array(cmp_init)
logger.info("different raw data shapes: %s vs %s", ref_data.shape, cmp_data.shape)
diffs += 1
elif len(ref_init.raw_data) > 0 and len(cmp_init.raw_data) > 0:
ref_data = to_array(ref_init)