Skip to content

[export_cpp] 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.