Skip to content

Add the training mode for the BatchNorm operator

Benjamin Halimi requested to merge bn_flag_redo into dev

Description

This MR aims to add the support of the training mode for the BatchNorm operator.

Previous version was only relying on the inference mode during the forward pass.

This MR is paired with aidge_core!263 (merged)

file modified : BatchNormImpl.cpp

Merge request reports

Copyright © Eclipse Foundation, Inc. All Rights Reserved.     Privacy Policy | Terms of Use | Copyright Agent