Solved by Brando_Miranda in post #4 
              
              
                Solution is to use mdl.train() it uses batch statistics by itself: 

Also by default, during training this layer keeps running estimates of its computed mean and variance, which are then used for normalization during evaluation. The running estimates are kept with a default momentum of 0.1. 
If trac…