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

forward_attn_mask and config update

Browse Source
This commit is contained in:
Eren Golge
2019-06-06 11:14:20 +02:00
parent 127a6b68e0
commit 31fe02412c
7 changed files with 32 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/tacotron.py
+3 -2
View File
@@ -19,6 +19,7 @@ class Tacotron(nn.Module):
prenet_dropout=True,
forward_attn=False,
trans_agent=False,
forward_attn_mask=False,
location_attn=True,
separate_stopnet=True):
super(Tacotron, self).__init__()
@@ -30,8 +31,8 @@ class Tacotron(nn.Module):
self.encoder = Encoder(256)
self.decoder = Decoder(256, mel_dim, r, memory_size, attn_win,
attn_norm, prenet_type, prenet_dropout,
forward_attn, trans_agent, location_attn,
separate_stopnet)
forward_attn, trans_agent, forward_attn_mask,
location_attn, separate_stopnet)
self.postnet = PostCBHG(mel_dim)
self.last_linear = nn.Sequential(
nn.Linear(self.postnet.cbhg.gru_features * 2, linear_dim),
models/tacotron2.py
+3 -2
View File
@@ -18,6 +18,7 @@ class Tacotron2(nn.Module):
prenet_dropout=True,
forward_attn=False,
trans_agent=False,
forward_attn_mask=False,
location_attn=True,
separate_stopnet=True):
super(Tacotron2, self).__init__()
@@ -30,8 +31,8 @@ class Tacotron2(nn.Module):
self.encoder = Encoder(512)
self.decoder = Decoder(512, self.n_mel_channels, r, attn_win,
attn_norm, prenet_type, prenet_dropout,
forward_attn, trans_agent, location_attn,
separate_stopnet)
forward_attn, trans_agent, forward_attn_mask,
location_attn, separate_stopnet)
self.postnet = Postnet(self.n_mel_channels)
def shape_outputs(self, mel_outputs, mel_outputs_postnet, alignments):