wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-10 11:50:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

config

Browse Source
This commit is contained in:
Eren Golge
2018-05-02 04:56:35 -07:00
parent d2657cbf3a
commit a4561c5096
5 changed files with 14 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/tacotron.py
+1 -3
View File
@@ -18,7 +18,6 @@ class Tacotron(nn.Module):
self.embedding.weight.data.normal_(0, 0.3)
self.encoder = Encoder(embedding_dim)
self.decoder = Decoder(256, mel_dim, r)
self.stopnet = nn.Sequential(nn.Linear(80, 1), nn.Sigmoid())
self.postnet = CBHG(mel_dim, K=8, projections=[256, mel_dim])
self.last_linear = nn.Linear(mel_dim * 2, linear_dim)
@@ -28,12 +27,11 @@ class Tacotron(nn.Module):
# batch x time x dim
encoder_outputs = self.encoder(inputs)
# batch x time x dim*r
mel_outputs, alignments = self.decoder(
mel_outputs, alignments, stop_tokens = self.decoder(
encoder_outputs, mel_specs)
# Reshape
# batch x time x dim
mel_outputs = mel_outputs.view(B, -1, self.mel_dim)
stop_tokens = self.stopnet(mel_outputs).squeeze()
linear_outputs = self.postnet(mel_outputs)
linear_outputs = self.last_linear(linear_outputs)
return mel_outputs, linear_outputs, alignments, stop_tokens