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

Graves attention and setting attn type by config.json

Browse Source
This commit is contained in:
Eren Golge
2019-11-12 11:18:57 +01:00
parent 84d81b6579
commit adf9ebd629
7 changed files with 96 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils/generic_utils.py
+4
View File
@@ -287,6 +287,7 @@ def setup_model(num_chars, num_speakers, c):
decoder_output_dim=c.audio['num_mels'],
gst=c.use_gst,
memory_size=c.memory_size,
attn_type=c.attention_type,
attn_win=c.windowing,
attn_norm=c.attention_norm,
prenet_type=c.prenet_type,
@@ -295,6 +296,7 @@ def setup_model(num_chars, num_speakers, c):
trans_agent=c.transition_agent,
forward_attn_mask=c.forward_attn_mask,
location_attn=c.location_attn,
attn_K=c.attention_heads,
separate_stopnet=c.separate_stopnet,
bidirectional_decoder=c.bidirectional_decoder)
elif c.model.lower() == "tacotron2":
@@ -303,6 +305,7 @@ def setup_model(num_chars, num_speakers, c):
r=c.r,
postnet_output_dim=c.audio['num_mels'],
decoder_output_dim=c.audio['num_mels'],
attn_type=c.attention_type,
attn_win=c.windowing,
attn_norm=c.attention_norm,
prenet_type=c.prenet_type,
@@ -311,6 +314,7 @@ def setup_model(num_chars, num_speakers, c):
trans_agent=c.transition_agent,
forward_attn_mask=c.forward_attn_mask,
location_attn=c.location_attn,
attn_K=c.attention_heads,
separate_stopnet=c.separate_stopnet,
bidirectional_decoder=c.bidirectional_decoder)
return model