Update models.py

2023-08-23 16:59:59 +08:00
parent bfc844ae02
commit a9289bb302
1 changed files with 1 additions and 1 deletions
--- a/models.py
+++ b/models.py
@@ -670,7 +670,7 @@ 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 = self.dp(x, x_mask, g=g)
        l_length_dp = torch.sum((logw - logw_) ** 2, [1, 2]) / torch.sum(x_mask)  # for averaging