Function spectral_norm

inpaint/networks.py:497–501

(module, mode=True)

def spectral_norm(module, mode=True):
    if mode:
        return nn.utils.spectral_norm(module)

    return module