From bd7a1c177b16318dde154d3f22c7c133468582a0 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Wed, 7 Apr 2021 10:25:33 +0200 Subject: [PATCH] fix #419 --- TTS/vocoder/layers/losses.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/TTS/vocoder/layers/losses.py b/TTS/vocoder/layers/losses.py index d5769204..7c4fded0 100644 --- a/TTS/vocoder/layers/losses.py +++ b/TTS/vocoder/layers/losses.py @@ -216,7 +216,11 @@ class MelganFeatureLoss(nn.Module): def forward(self, fake_feats, real_feats): loss_feats = 0 num_feats = 0 +<<<<<<< HEAD for idx, _ in enumerate(fake_feats): +======= + for idx in range(len(fake_feats)): +>>>>>>> fix #419 for fake_feat, real_feat in zip(fake_feats[idx], real_feats[idx]): loss_feats += self.loss_func(fake_feat, real_feat) num_feats += 1 @@ -335,7 +339,7 @@ class GeneratorLoss(nn.Module): if self.use_l1_spec_loss: l1_spec_loss = self.l1_spec_loss(y_hat, y) return_dict['G_l1_spec_loss'] = l1_spec_loss - gen_loss += self.l1_spec_loss_weight * l1_spec_loss + gen_loss = gen_loss + self.l1_spec_loss_weight * l1_spec_loss # subband STFT Loss if self.use_subband_stft_loss: