diff --git a/examples/export_ResNet18/resnet18.py b/examples/export_ResNet18/resnet18.py index f60e68ae1b7df4fcbeb1f52a0808b839f1ed5633..64b3a9b18b7ca6e0db465b5491aa3ee2a3c5e117 100644 --- a/examples/export_ResNet18/resnet18.py +++ b/examples/export_ResNet18/resnet18.py @@ -502,8 +502,9 @@ This option has to be passed to each node in order to be used within the Export (JConv, JPad, ...) that you can find in the "export_gen/operator_export" folder. """ -for node in model.get_nodes(): - node.attributes().aidge_cmp = AIDGE_CMP +if AIDGE_CMP: + for node in model.get_nodes(): + node.attributes().aidge_cmp = True # -------------------------------------------------------------- # EXPORT THE MODEL