Skip to content
Snippets Groups Projects

feat/script_compare_layers_aidge_onnx

All threads resolved!
3 files
+ 12
35
Compare changes
  • Side-by-side
  • Inline
Files
3
@@ -193,6 +193,7 @@ def compare_layers_aidge_onnx(
aidge_nodes = [
node for node in aidge_model.get_nodes() if node.name() in aidge_node_names
]
# Iterate over all nodes and compare their outputs
all_layers_identical = True
layers_error_above_threshold = []
@@ -203,10 +204,6 @@ def compare_layers_aidge_onnx(
continue
aidge_node_to_search = onnx_to_aidge_name(name)
aidge_node = find_node_with_name(aidge_nodes, aidge_node_to_search)
if aidge_node == -1:
aidge_node = find_node_with_name(
aidge_model.get_output_nodes(), f"data_{aidge_node_to_search}"
)
if aidge_node == -1:
print(
f"ERROR : no aidge node found with name {aidge_node_to_search}"
Loading