Skip to content

Add the support of the training_mode for the BatchNorm operator

Benjamin Halimi requested to merge bn_flag into dev

Description

This minor MR intends to add the support of the training_mode for BatchNorm operator.

files modified : node_export/aidge_converters/batchnorm.py, node_import/onnx_converters/batchnorm.py

Merge request reports

Loading