regularizing the softmax weights hopefully fixes the bug (softmax saturation)