Is that a valid solution? How can you check if Conv-layer is equiped with spectral norm?
# Make all layers to be spectral normalization layer
def add_sn(m):    
    if isinstance(m, (nn.Conv2d, nn.ConvTranspose2d)):
        return spectral_norm(m)
    else:
        return m
    
Generator.apply(add_sn)