Merge pull request #5 from ylzz1997/master

反量化
2023-08-30 00:58:50 +08:00
parent 4874c27349 c3e442ab9c
commit 880f56d46d
2 changed files with 26532 additions and 0 deletions
--- a/models.py
+++ b/models.py
@@ -665,6 +665,8 @@ class SynthesizerTrn(nn.Module):
            attn = monotonic_align.maximum_path(neg_cent, attn_mask.squeeze(1)).unsqueeze(1).detach()

        w = attn.sum(2)
+        # 反量化
+        w -= torch.rand_like(w)

        l_length_sdp = self.sdp(x, x_mask, w, g=g)
        l_length_sdp = l_length_sdp / torch.sum(x_mask)
--- a/monotonic_align/core.c
+++ b/monotonic_align/core.c