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

Update models.py

Browse Source
This commit is contained in:
Stardust·减
2023-08-21 08:32:39 +08:00
committed by GitHub
parent d1416dc096
commit 1a223661ab
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
models.py
View File

@@ -182,7 +182,8 @@ class TextEncoder(nn.Module):
n_heads, n_heads,
n_layers, n_layers,
kernel_size, kernel_size,
p_dropout): p_dropout,
gin_channels=0):
super().__init__() super().__init__()
self.n_vocab = n_vocab self.n_vocab = n_vocab
self.out_channels = out_channels self.out_channels = out_channels
@@ -192,7 +193,7 @@ class TextEncoder(nn.Module):
self.n_layers = n_layers self.n_layers = n_layers
self.kernel_size = kernel_size self.kernel_size = kernel_size
self.p_dropout = p_dropout self.p_dropout = p_dropout
self.gin_channels = gin_channels
self.emb = nn.Embedding(len(symbols), hidden_channels) self.emb = nn.Embedding(len(symbols), hidden_channels)
nn.init.normal_(self.emb.weight, 0.0, hidden_channels ** -0.5) nn.init.normal_(self.emb.weight, 0.0, hidden_channels ** -0.5)
self.tone_emb = nn.Embedding(num_tones, hidden_channels) self.tone_emb = nn.Embedding(num_tones, hidden_channels)
@@ -207,7 +208,8 @@ class TextEncoder(nn.Module):
n_heads, n_heads,
n_layers, n_layers,
kernel_size, kernel_size,
p_dropout) p_dropout,
gin_channels=self.gin_channels)
self.proj = nn.Conv1d(hidden_channels, out_channels * 2, 1) self.proj = nn.Conv1d(hidden_channels, out_channels * 2, 1)
def forward(self, x, x_lengths, tone, language, bert): def forward(self, x, x_lengths, tone, language, bert):