def apply_weight_norm()

in parler_tts/dac_wrapper/modeling_dac.py [0:0]

• 8 lines of code
• 4 McCabe index (conditional complexity)

    def apply_weight_norm(self):
        weight_norm = nn.utils.weight_norm
        if hasattr(nn.utils.parametrizations, "weight_norm"):
            weight_norm = nn.utils.parametrizations.weight_norm

        def _apply_weight_norm(module):
            if isinstance(module, nn.Conv1d) or isinstance(module, nn.ConvTranspose1d):
                weight_norm(module)

        self.apply(_apply_weight_norm)