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

bug fix

This commit is contained in:
Eren Golge 2018-12-12 12:02:10 +01:00
parent 22d62aee5b
commit 703be04993
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/tacotron.py
View File

@ -338,8 +338,8 @@ class Decoder(nn.Module):
gain=torch.nn.init.calculate_gain('linear')) gain=torch.nn.init.calculate_gain('linear'))
def _reshape_memory(self, memory): def _reshape_memory(self, memory):
B = memory.shape[0]
if memory is not None: if memory is not None:
B = memory.shape[0]
# Grouping multiple frames if necessary # Grouping multiple frames if necessary
if memory.size(-1) == self.memory_dim: if memory.size(-1) == self.memory_dim:
memory = memory.contiguous() memory = memory.contiguous()