fix: store weights after normalization