From 8ed9f57a6d26c7fb67f0a8f661b54c8660eb55c6 Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Fri, 11 May 2018 04:19:28 -0700 Subject: [PATCH] bug fix --- layers/tacotron.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/layers/tacotron.py b/layers/tacotron.py index 756d9b26..1304fa74 100644 --- 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: