Modification of the forward method of the batchnorm.hpp file to match the...
Context
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
.
Issue: #47
Modified files
(Summarize the changes you made for each modified file - example:
-
batchnorm.hpp
, update forward method;