[TMVA][SOFIE] Add batch normalization operator#9307
[TMVA][SOFIE] Add batch normalization operator#9307lmoneta merged 3 commits intoroot-project:masterfrom
Conversation
…imensions less than 4. For example this is the case when using models created with torch BatchNorm1d (Batch normalization for dense layers)
lmoneta
requested review from
ashlaban,
bellenot,
sitongan and
stwunsch
as code owners
|
Starting build on |
|
@phsft-bot build just on ROOT-ubuntu18.04/default with flags -Dtmva-sofie=On |
|
Starting build on |
Fix 1d Batch normalization. Do not assume tensor that the input is of size 4 and tensor output is of size 4 Fix a warning message in Sofie parser for batchnorm operator Fix linking of Blas in TestSofieModels
lmoneta
force-pushed
the
tmva_sofie_batch-normalization
branch
from
7972484 to
1d4cdd2
Compare
|
Starting build on |
|
@phsft-bot build just on ROOT-ubuntu18.04/default with flags -Dtmva-sofie=On |
|
Starting build on |
|
@lmoneta for future commits, can you please leave an empty line between the subject (first line) and the commit description (starting in line 3)? Having text in the second line makes |
|
You are right ! Sorry for this, I have never used |
5 participants
Add the new operator developed by Aaradhya Saxena.
This PR extends #8882 and updates to the new Session class of Sofie.
It includes also the test for Batch normalization operator