From 460d0675a6657f4037d38340ed7b997eb90ef593 Mon Sep 17 00:00:00 2001 From: erogol Date: Mon, 15 Jun 2020 11:47:59 +0200 Subject: [PATCH] bug fix for vocoder feature macthing loss --- vocoder/layers/losses.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vocoder/layers/losses.py b/vocoder/layers/losses.py index 4774ab4e..22237077 100644 --- a/vocoder/layers/losses.py +++ b/vocoder/layers/losses.py @@ -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