mirror of https://github.com/coqui-ai/TTS.git
Change to GMMv2b
This commit is contained in:
parent
34e0291ba7
commit
748cbbc403
|
@ -161,7 +161,6 @@ class GravesAttention(nn.Module):
|
||||||
# attention GMM parameters
|
# attention GMM parameters
|
||||||
sig_t = torch.nn.functional.softplus(b_t)+self.eps
|
sig_t = torch.nn.functional.softplus(b_t)+self.eps
|
||||||
|
|
||||||
#inv_sig_t = torch.exp(-torch.clamp(b_t, min=-6, max=9)) # variance
|
|
||||||
mu_t = self.mu_prev + torch.nn.functional.softplus(k_t)
|
mu_t = self.mu_prev + torch.nn.functional.softplus(k_t)
|
||||||
g_t = torch.softmax(g_t, dim=-1) / sig_t + self.eps
|
g_t = torch.softmax(g_t, dim=-1) / sig_t + self.eps
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue