From 8fde0ac00ec34197ab1a94539bf835a8c4be6ebc Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Fri, 16 Aug 2019 14:23:26 +0200 Subject: [PATCH] pylint --- layers/tacotron.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/layers/tacotron.py b/layers/tacotron.py index d8d0e57a..09c7e923 100644 --- a/layers/tacotron.py +++ b/layers/tacotron.py @@ -397,9 +397,10 @@ class Decoder(nn.Module): if self.memory_size > self.r: # memory queue size is larger than number of frames per decoder iter 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: # memory queue size smaller than number of frames per decoder iter self.memory_input = new_memory[:, :self.memory_size * self.memory_dim]