mirror of https://github.com/coqui-ai/TTS.git
larger attention filter size and mode filters
This commit is contained in:
parent
56c6d0cac8
commit
f377cd3cb8
|
@ -37,8 +37,8 @@ class LocationSensitiveAttention(nn.Module):
|
||||||
annot_dim,
|
annot_dim,
|
||||||
query_dim,
|
query_dim,
|
||||||
attn_dim,
|
attn_dim,
|
||||||
kernel_size=7,
|
kernel_size=31,
|
||||||
filters=20):
|
filters=32):
|
||||||
super(LocationSensitiveAttention, self).__init__()
|
super(LocationSensitiveAttention, self).__init__()
|
||||||
self.kernel_size = kernel_size
|
self.kernel_size = kernel_size
|
||||||
self.filters = filters
|
self.filters = filters
|
||||||
|
|
Loading…
Reference in New Issue