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

pylint

This commit is contained in:
Eren Golge 2019-08-16 14:23:26 +02:00
parent 728b97da3a
commit 8fde0ac00e
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
layers/tacotron.py
View File

@ -397,9 +397,10 @@ class Decoder(nn.Module):
if self.memory_size > self.r: if self.memory_size > self.r:
# memory queue size is larger than number of frames per decoder iter # memory queue size is larger than number of frames per decoder iter
self.memory_input = torch.cat([ self.memory_input = torch.cat([
new_memory, self.memory_input[:, :(self.memory_size - self.r) * self.memory_dim].clone() new_memory, self.memory_input[:, :(
self.memory_size - self.r) * self.memory_dim].clone()
], ],
dim=-1) dim=-1)
else: else:
# memory queue size smaller than number of frames per decoder iter # memory queue size smaller than number of frames per decoder iter
self.memory_input = new_memory[:, :self.memory_size * self.memory_dim] self.memory_input = new_memory[:, :self.memory_size * self.memory_dim]