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

bug fix for vocoder feature macthing loss

pull/1/head
erogol 2020-06-15 11:47:59 +02:00
parent 98fd3b53f2
commit 460d0675a6
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
vocoder/layers/losses.py
View File

@ -139,7 +139,7 @@ class MelganFeatureLoss(nn.Module):
def forward(self, fake_feats, real_feats):
loss_feats = 0
for fake_feat, real_feat in zip(fake_feats, real_feats):
loss_feats += self.loss_func(fake_feats, real_feats)
loss_feats += self.loss_func(fake_feat, real_feat)
loss_feats /= len(fake_feats) + len(real_feats)
return loss_feats