From a25ba81296d2727d17fdf7037f1a4463e0b10e99 Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Fri, 25 May 2018 04:28:40 -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 e3bcf655..041785d2 100644 --- a/layers/tacotron.py +++ b/layers/tacotron.py @@ -200,7 +200,7 @@ class Decoder(nn.Module): def __init__(self, in_features, memory_dim, r): super(Decoder, self).__init__() - self.mode = mode + self.r = r self.max_decoder_steps = 200 self.memory_dim = memory_dim # memory -> |Prenet| -> processed_memory