bug fix for vocoder feature macthing loss

This commit is contained in:
erogol
2020-06-15 19:04:05 +02:00
parent 98fd3b53f2
commit 460d0675a6
+1 -1
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