def remove_weight_norm()

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

• 5 lines of code
• 3 McCabe index (conditional complexity)

    def remove_weight_norm(self):
        def _remove_weight_norm(module):
            if isinstance(module, nn.Conv1d) or isinstance(module, nn.ConvTranspose1d):
                nn.utils.remove_weight_norm(module)
        self.apply(_remove_weight_norm)