MycroftAI
/
mimic2
mirror of https://github.com/MycroftAI/mimic2.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

restore previous attention cell implemention

test
Michael Nguyen 2018-09-14 14:13:46 -05:00
parent 90b332044e
commit 0354b95af5
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
models/tacotron.py
View File

@ -49,12 +49,11 @@ class Tacotron():
# Attention
attention_cell = AttentionWrapper(
GRUCell(256),
DecoderPrenetWrapper(GRUCell(256), is_training),
LocationSensitiveAttention(256, encoder_outputs),
alignment_history=True,
output_attention=False) # [N, T_in, 256]
attention_cell = DecoderPrenetWrapper(attention_cell, is_training)
# Concatenate attention context vector and RNN cell output into a 512D vector.
concat_cell = ConcatOutputAndAttentionWrapper(attention_cell) # [N, T_in, 512]