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