short final minibatch seems ok now; also adjusted the critical regions for normalization_cache