I tried morph net with grouplasso, but my trained network report error with layer normalization, how to solve it?