From 6a70c63f242c49f2945d1f97b71efb0cdfb138c2 Mon Sep 17 00:00:00 2001
From: erogol <erogol@hotmail.com>
Date: Sun, 27 Sep 2020 03:28:42 +0200
Subject: [PATCH] correct glow-tts loss

---
 TTS/tts/layers/losses.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/TTS/tts/layers/losses.py b/TTS/tts/layers/losses.py
index bf03671c..15b0a2ee 100644
--- a/TTS/tts/layers/losses.py
+++ b/TTS/tts/layers/losses.py
@@ -270,7 +270,7 @@ class GlowTTSLoss(torch.nn.Module):
         pz = torch.sum(scales) + 0.5 * torch.sum(
             torch.exp(-2 * scales) * (z - means)**2)
         log_mle = self.constant_factor + (pz - torch.sum(log_det)) / (
-            torch.sum(y_lengths // 2) * 2 * z.shape[1])
+            torch.sum(y_lengths) * z.shape[1])
         # duration loss - MSE
         # loss_dur = torch.sum((o_dur_log - o_attn_dur)**2) / torch.sum(x_lengths)
         # duration loss - huber loss