fix(scripts): show data shape in initializer diff
This commit is contained in:
parent
512f41135d
commit
b4095cc2db
|
@ -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)
|
logger.info("different data types: %s vs %s", ref_init.data_type, cmp_init.data_type)
|
||||||
diffs += 1
|
diffs += 1
|
||||||
elif len(ref_init.raw_data) != len(cmp_init.raw_data):
|
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
|
diffs += 1
|
||||||
elif len(ref_init.raw_data) > 0 and len(cmp_init.raw_data) > 0:
|
elif len(ref_init.raw_data) > 0 and len(cmp_init.raw_data) > 0:
|
||||||
ref_data = to_array(ref_init)
|
ref_data = to_array(ref_init)
|
||||||
|
|
Loading…
Reference in New Issue