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

Correct commnet

This commit is contained in:
Eren Golge 2018-05-18 03:32:17 -07:00
parent d43b4ccc1e
commit 7b9fd63649
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
layers/attention.py
View File

@ -13,8 +13,8 @@ class BahdanauAttention(nn.Module):
def forward(self, annots, query): def forward(self, annots, query):
""" """
Shapes: Shapes:
- query: (batch, 1, dim) or (batch, dim)
- annots: (batch, max_time, dim) - annots: (batch, max_time, dim)
- query: (batch, 1, dim) or (batch, dim)
""" """
if query.dim() == 2: if query.dim() == 2:
# insert time-axis for broadcasting # insert time-axis for broadcasting