Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

remove weight norm for melgan generator

This commit is contained in:
erogol 2020-06-09 23:02:28 +02:00
parent f9698af06e
commit c866f23af6
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
vocoder/models/melgan_generator.py
View File

@ -87,3 +87,12 @@ class MelganGenerator(nn.Module):
(self.inference_padding, self.inference_padding), (self.inference_padding, self.inference_padding),
'replicate') 'replicate')
return self.layers(cond_features) return self.layers(cond_features)
def remove_weight_norm(self):
for _, layer in enumerate(self.layers):
if len(layer.state_dict()) != 0:
try:
nn.utils.remove_weight_norm(layer)
except ValueError:
layer.remove_weight_norm()