tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Merge pull request #6 from ylzz1997/master

Browse Source 
debug 反量化 我发现flow里面有反量化的代码
This commit is contained in:
Stardust·减
2023-08-30 08:50:47 +08:00
committed by GitHub
parent 880f56d46d 6b9f4d11d2
commit fef5f41aec
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
models.py
View File

@@ -665,12 +665,13 @@ class SynthesizerTrn(nn.Module):
attn = monotonic_align.maximum_path(neg_cent, attn_mask.squeeze(1)).unsqueeze(1).detach() attn = monotonic_align.maximum_path(neg_cent, attn_mask.squeeze(1)).unsqueeze(1).detach()
w = attn.sum(2) w = attn.sum(2)
# 反量化
w -= torch.rand_like(w)
l_length_sdp = self.sdp(x, x_mask, w, g=g) l_length_sdp = self.sdp(x, x_mask, w, g=g)
l_length_sdp = l_length_sdp / torch.sum(x_mask) l_length_sdp = l_length_sdp / torch.sum(x_mask)
# 反量化
w -= torch.rand_like(w)
logw_ = torch.log(w + 1e-6) * x_mask logw_ = torch.log(w + 1e-6) * x_mask
logw = self.dp(x, x_mask, g=g) logw = self.dp(x, x_mask, g=g)
l_length_dp = torch.sum((logw - logw_) ** 2, [1, 2]) / torch.sum(x_mask) # for averaging l_length_dp = torch.sum((logw - logw_) ** 2, [1, 2]) / torch.sum(x_mask) # for averaging