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

add bias to attention v

This commit is contained in:
Eren Golge 2019-03-19 12:21:36 +01:00
parent 1898763028
commit 82cde95cfa
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/tacotron2.py
View File

@ -118,7 +118,7 @@ class Attention(nn.Module):
attention_rnn_dim, attention_dim, bias=False, init_gain='tanh') attention_rnn_dim, attention_dim, bias=False, init_gain='tanh')
self.inputs_layer = Linear( self.inputs_layer = Linear(
embedding_dim, attention_dim, bias=False, init_gain='tanh') embedding_dim, attention_dim, bias=False, init_gain='tanh')
self.v = Linear(attention_dim, 1, bias=False) self.v = Linear(attention_dim, 1, bias=True)
self.location_layer = LocationLayer(attention_location_n_filters, self.location_layer = LocationLayer(attention_location_n_filters,
attention_location_kernel_size, attention_location_kernel_size,
attention_dim) attention_dim)