Should be normalizing gradients, not weights.