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

Make attn windowing optional

Browse Source
This commit is contained in:
Eren Golge
2019-01-16 12:33:07 +01:00
parent 8969d59902
commit f4fa155cd3
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
layers/attention.py
+1 -2
View File
@@ -160,8 +160,7 @@ class AttentionRNNCell(nn.Module):
mask = mask.view(memory.size(0), -1)
alignment.masked_fill_(1 - mask, -float("inf"))
# Windowing
if not self.training:
# print(" > Windowing active")
if not self.training and self.windowing:
back_win = self.win_idx - self.win_back
front_win = self.win_idx + self.win_front
if back_win > 0: