After successful first stage training, the generated samples suddenly become completely grey as if all neurons died, when a norm_fn is selected.
Leads to investigate: transfer of weights to the extended networks, parameter freezing
Last samples of first stage (16x16):

First samples of the second stage (32x32):
