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): 