Skip to content

Modification of the forward method of the batchnorm.hpp file to match the...

Clément Fisher requested to merge clementgf/aidge_export_cpp:batchnorm-fix into dev

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;

Merge request reports

Loading