From df34d6f78a9f2a19af29071af069bd49fbcabed6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= <2225664821@qq.com>
Date: Tue, 29 Aug 2023 20:48:05 +0800
Subject: [PATCH] Update models.py

---
 models.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/models.py b/models.py
index 9951bf6..6052a76 100644
--- a/models.py
+++ b/models.py
@@ -669,7 +669,8 @@ class SynthesizerTrn(nn.Module):
         l_length_sdp = self.sdp(x, x_mask, w, g=g)
         l_length_sdp = l_length_sdp / torch.sum(x_mask)
         
-        logw_ = torch.log(w + 1e-6) * x_mask
+        #logw_ = torch.log(w + 1e-6) * x_mask
+        logw_ = torch.log((w-torch.rand_like(w)) * x_mask + 1e-6) * x_mask
         logw = self.dp(x, x_mask, g=g)
         l_length_dp = torch.sum((logw - logw_) ** 2, [1, 2]) / torch.sum(x_mask)  # for averaging