From a68487f0b8fb60605de96b7b0a8b3925958fb83e Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Mon, 26 Mar 2018 11:08:15 -0700 Subject: [PATCH] Data loader bug fix 3 --- layers/tacotron.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/layers/tacotron.py b/layers/tacotron.py index 9ec8bffb..916ea677 100644 --- a/layers/tacotron.py +++ b/layers/tacotron.py @@ -256,13 +256,10 @@ class Decoder(nn.Module): greedy = not self.training if memory is not None: - print(memory.shape) # Grouping multiple frames if necessary if memory.size(-1) == self.memory_dim: memory = memory.view(B, memory.size(1) // self.r, -1) - print(memory.shape) - assert memory.size(-1) == self.memory_dim * self.r,\ " !! Dimension mismatch {} vs {} * {}".format(memory.size(-1), self.memory_dim, self.r) T_decoder = memory.size(1)