diff --git a/models.py b/models.py index 58af3da..8e6084d 100644 --- 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