bug fix

2018-05-11 04:19:28 -07:00 · 2018-05-11 04:19:28 -07:00 · 8ed9f57a6d
parent 31cc9c5443
commit 8ed9f57a6d
1 changed files with 1 additions and 1 deletions
--- a/layers/tacotron.py
+++ b/layers/tacotron.py
@ -303,7 +303,7 @@ class Decoder(nn.Module):
            stop_token = self.stopnet(output)
            outputs += [output]
            alignments += [alignment]
-            stop_tokens += stop_token
+            stop_tokens += [stop_token]
            t += 1
            if (not greedy and self.training) or (greedy and memory is not None):
                if t >= T_decoder: