Problem when exporting models with a BatchNorm layer
Required prerequisites
-
Make sure you've read the documentation. Your issue may be addressed there. -
Search the issue tracker and discussions to verify that this hasn't already been reported. +1 or comment there if it has.
What commit version of aidge do you use
-
aidge_core
: 0.7.0 -
aidge_export_cpp
: 0.4.0
Problem description
The output for exported models containing a BatchNorm layer appears to be incorrect, even for a minimal model consisting of a single BatchNorm layer. In the C++ code, the forward
function expects the values in the format [inputs, outputs, biases, variances, means, scales]
. However, after enabling intermediate output file saving in the MakeFile (SAVE_FILE
), it was observed that the data is structured as [inputs, scale, biases, mean, variances]
, which aligns with the definition of the BatchNorm producers in aidge_core
.