From 7d0ec62bf1f619d0729507738849fc82553145f0 Mon Sep 17 00:00:00 2001 From: shaun <17098117+shaun95@users.noreply.github.com> Date: Sun, 2 May 2021 14:14:24 +0200 Subject: [PATCH] Update losses.py The block of code for use_l1_spec_loss is repeated which doubles the amount of L1 loss when enabled. The weight for L1 loss in hifigan_ljspeech configutation will likely need to be doubled to compensate (l1_spec_loss_weight) --- TTS/vocoder/layers/losses.py | 6 ------ 1 file changed, 6 deletions(-) diff --git a/TTS/vocoder/layers/losses.py b/TTS/vocoder/layers/losses.py index e291a112..2e204115 100644 --- a/TTS/vocoder/layers/losses.py +++ b/TTS/vocoder/layers/losses.py @@ -350,12 +350,6 @@ class GeneratorLoss(nn.Module): return_dict["G_l1_spec_loss"] = l1_spec_loss gen_loss = gen_loss + self.l1_spec_loss_weight * l1_spec_loss - # L1 Spec loss - 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 = gen_loss + self.l1_spec_loss_weight * l1_spec_loss - # subband STFT Loss if self.use_subband_stft_loss: subband_stft_loss_mg, subband_stft_loss_sc = self.subband_stft_loss(y_hat_sub, y_sub)