coqui-ai
/
TTS
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Configurable alignment method

pull/10/head
Eren Golge 2018-05-23 06:17:01 -07:00
parent d8c460442a
commit 0f933106ca
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
layers/attention.py
View File

@ -105,7 +105,7 @@ class AttentionRNN(nn.Module):
# Alignment # Alignment
# (batch, max_time) # (batch, max_time)
# e_{ij} = a(s_{i-1}, h_j) # e_{ij} = a(s_{i-1}, h_j)
if attnetion_vec is None: if self.align_model is 'b':
alignment = self.alignment_model(annotations, rnn_output) alignment = self.alignment_model(annotations, rnn_output)
else: else:
alignment = self.alignment_model(annotations, rnn_output, attention_vec) alignment = self.alignment_model(annotations, rnn_output, attention_vec)