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

Attention biased chaged

This commit is contained in:
Eren Golge 2019-01-22 18:18:21 +01:00
parent 83d6f1dcee
commit 66f8d0e260
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
layers/attention.py
View File

@ -52,9 +52,9 @@ class LocationSensitiveAttention(nn.Module):
stride=1,
padding=0,
bias=False))
self.loc_linear = nn.Linear(filters, attn_dim, bias=False)
self.query_layer = nn.Linear(query_dim, attn_dim, bias=False)
self.annot_layer = nn.Linear(annot_dim, attn_dim, bias=False)
self.loc_linear = nn.Linear(filters, attn_dim, bias=True)
self.query_layer = nn.Linear(query_dim, attn_dim, bias=True)
self.annot_layer = nn.Linear(annot_dim, attn_dim, bias=True)
self.v = nn.Linear(attn_dim, 1, bias=True)
self.processed_annots = None
# self.init_layers()