Merge branch 'master' of github.com:mozilla/TTS

2018-12-12 15:45:48 +01:00 · 2018-12-12 15:45:48 +01:00 · 3a72d75ecd
parent 8d865629a0 703be04993
commit 3a72d75ecd
1 changed files with 1 additions and 1 deletions
--- a/layers/tacotron.py
+++ b/layers/tacotron.py
@ -371,7 +371,7 @@ class Decoder(nn.Module):
        greedy = not self.training
        if memory is not None:
            memory = self._reshape_memory(memory)
-        T_decoder = memory.size(0)
+            T_decoder = memory.size(0)
        # go frame as zeros matrix
        initial_memory = inputs.data.new(B, self.memory_dim * self.r).zero_()
        # decoder states