Great! And thank you for the script along with it. Just to confirm, this is the only change moving from train to eval mode for batch normalization will have? That is, there isn’t another part of the batch normalization that will change when using eval mode?