diff --git a/aidge_export_cpp/operators.py b/aidge_export_cpp/operators.py index de6c9642c66f3ae92a022cacbcb8e9b99ea056d7..c275a085fcc0dadfbf46faebf3568dc47e71fc0d 100644 --- a/aidge_export_cpp/operators.py +++ b/aidge_export_cpp/operators.py @@ -154,7 +154,7 @@ class ConvCPP(ExportNode): CONFIGURATIONS.CONV, name=self.name, input_dims=self.inputs_dims[0][1:], - output_dims=self.outputs_dims[0], + output_dims=self.outputs_dims[0][1:], kernel=self.kernel, stride=self.stride, padding=self.padding,