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

set bias

This commit is contained in:
Eren Golge 2019-01-16 15:53:24 +01:00
parent 915783e10e
commit af22bed149
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
layers/attention.py
View File

@ -52,10 +52,10 @@ class LocationSensitiveAttention(nn.Module):
stride=1,
padding=0,
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=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.v = nn.Linear(attn_dim, 1, bias=True)
self.processed_annots = None
# self.init_layers()