Fix the batch normalization updates

2 jobs for amir in 47 minutes and 17 seconds (queued for 8 seconds)