From de3a04f104527cf5a164fefd3221a5d144c4e652 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Tue, 6 Apr 2021 11:02:16 +0200 Subject: [PATCH] some commeting for Generator loss and check if the argument is defines in the config file --- TTS/vocoder/layers/losses.py | 6 ++++++ 1 file changed, 6 insertions(+) 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)