wassname/TTS
1
0 Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-10-29 12:51:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

set bias

Browse Source
This commit is contained in:
Eren Golge committed 2019-01-16 15:53:24 +01:00
1 parent 915783e10e
commit af22bed149
1 file changed
+4 -4
Download Patch File Download Diff File Expand all files Collapse all files
layers/attention.py
+4 -4
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()