diff --git a/TTS/vocoder/layers/losses.py b/TTS/vocoder/layers/losses.py index ab6f3d35..d5769204 100644 --- a/TTS/vocoder/layers/losses.py +++ b/TTS/vocoder/layers/losses.py @@ -331,6 +331,12 @@ 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 += 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)