diff --git a/aidge_export_cpp/operators/Producer.py b/aidge_export_cpp/operators/Producer.py index 1ae54d3b3675d112ba89e9c1687d5745d5218c54..138b58cc049a853bb0420e2e511ef27496436258 100644 --- a/aidge_export_cpp/operators/Producer.py +++ b/aidge_export_cpp/operators/Producer.py @@ -47,7 +47,10 @@ class ProducerCPP(ExportNode): def __init__(self, node, mem_info): super().__init__(node, mem_info) self.values = np.array(self.operator.get_output(0)) - self.ignore = node.attributes().ignore + if node.attributes().has_attr("ignore"): + self.ignore = node.attributes().ignore + else: + self.ignore = False if len(self.values.shape) == 4: # Note: export in HWC self.values = np.transpose(self.values, (0, 2, 3, 1))